Home
last modified time | relevance | path

Searched refs:stepOperandIndex (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionPlan.cpp421 *stepOperandIndex = i->second; in addOperand()
426 *stepOperandIndex = mStepModel.operandCount(); in addOperand()
427 mOperandMap.emplace(sourceOperandIndex, *stepOperandIndex); in addOperand()
446 n = copyOperandExtraParams(mStepModel, *stepOperandIndex, operand); in addOperand()
464 n = mStepModel.setOperandValue(*stepOperandIndex, nullptr, 0); in addOperand()
481 mModelInputs.emplace_back(sourceOperandIndex, *stepOperandIndex); in addOperand()
492 mModelOutputs.emplace_back(sourceOperandIndex, *stepOperandIndex); in addOperand()
501 n = mStepModel.setOperandValueFromModel(*stepOperandIndex, model); in addOperand()
686 void ExecutionStep::recordTempAsStepModelOutput(uint32_t stepOperandIndex) { in recordTempAsStepModelOutput() argument
687 const auto it = mOperandMap.find(stepOperandIndex); in recordTempAsStepModelOutput()
[all …]
H A DExecutionPlan.h242 int addOperand(uint32_t sourceOperandIndex, uint32_t* stepOperandIndex, OperandKind kind);
272 void recordTempAsStepModelOutput(uint32_t stepOperandIndex);