Searched refs:mOperands (Results 1 - 5 of 5) sorted by relevance

/frameworks/ml/nn/common/
H A DCpuExecutor.cpp161 mOperands.resize(count);
166 RunTimeOperandInfo& to = mOperands[i];
210 RunTimeOperandInfo& to = mOperands[operandIndex];
238 auto& info = mOperands[i];
275 if (mOperands[indexes[i]].lifetime == OperandLifeTime::NO_VALUE) {
295 const RunTimeOperandInfo& in1 = mOperands[ins[0]];
296 const RunTimeOperandInfo& in2 = mOperands[ins[1]];
297 int32_t activation = getScalarData<int32_t>(mOperands[ins[2]]);
299 RunTimeOperandInfo& out = mOperands[outs[0]];
328 const RunTimeOperandInfo& in1 = mOperands[in
[all...]
/frameworks/ml/nn/runtime/
H A DModelBuilder.h60 return static_cast<uint32_t>(mOperands.size());
70 return mOperands[getInputOperandIndex(i)];
74 return mOperands[getOutputOperandIndex(i)];
76 const Operand& getOperand(uint32_t index) const { return mOperands[index]; }
109 std::vector<Operand> mOperands; member in class:android::nn::ModelBuilder
H A DModelBuilder.cpp43 size_t idx = mOperands.size();
48 mOperands.resize(idx + 1);
49 auto& operand = mOperands[idx];
67 Operand& operand = mOperands[index];
121 Operand& operand = mOperands[l.operandIndex];
144 Operand& operand = mOperands[l.operandIndex];
160 Operand& operand = mOperands[index];
208 mOperands[i].numberOfConsumers++;
209 // TODO mOperands[i].consumers.push_back(operationIndex);
238 if (operandIndex >= mOperands
[all...]
H A DExecutionPlan.cpp768 const OperandType operandType = mOperands[operandIndex].type;
/frameworks/ml/nn/common/include/
H A DCpuExecutor.h106 std::vector<RunTimeOperandInfo> mOperands; member in class:android::nn::CpuExecutor

Completed in 79 milliseconds