Searched refs:mSubModel (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.cpp | 133 : mPlan(plan), mIndex(stepIndex), mSubModel(), mDevice(device) {} in ExecutionStep()
148 *toOperandIndex = mSubModel.operandCount(); in addOperand()
160 int n = mSubModel.addOperand(type); in addOperand()
170 n = mSubModel.setOperandValue(*toOperandIndex, data, operand.location.length); in addOperand()
178 n = mSubModel.setOperandValueFromMemory(*toOperandIndex, memory, in addOperand()
187 n = mSubModel.setOperandValue(*toOperandIndex, nullptr, 0); in addOperand()
265 return mSubModel.addOperation(static_cast<uint32_t>(operation.type), inputCount, inputs.data(), in addOperation()
355 mSubModel.relaxComputationFloat32toFloat16(fromModel->isComputationFloat32RelaxedToFloat16()); in finishSubModel()
383 const Operand& operand = mSubModel.getOperand(subModelOutput.second); in finishSubModel()
396 …int n = mSubModel.identifyInputsAndOutputs(inputs.size(), &inputs[0], outputs.size(), &outputs[0]); in finishSubModel()
[all …]
|
| D | ExecutionPlan.h | 86 const ModelBuilder* getSubModel() const { return &mSubModel; } in getSubModel()
106 ModelBuilder mSubModel; variable
|