Home
last modified time | relevance | path

Searched refs:sourceOperandIndex (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionPlan.cpp190 std::string toString(SourceOperandIndex sourceOperandIndex) { in toString() argument
192 std::to_string(sourceOperandIndex.second) + ")"; in toString()
308 << toString(sourceOperandIndex) in redeclare()
418 auto i = mOperandMap.find(sourceOperandIndex); in addOperand()
586 << toString(sourceOperandIndex); in mapInputsAndOutputs()
604 << toString(sourceOperandIndex); in mapInputsAndOutputs()
658 recordAsOutputIfTemporary(sourceOperandIndex); in findTempsAsStepModelOutputs()
663 recordAsOutputIfTemporary(sourceOperandIndex); in findTempsAsStepModelOutputs()
883 for (const auto& sourceOperandIndex : operands) { in finish() local
995 handleBoundaryConstants(sourceOperandIndex); in findControlFlowBoundaryConstants()
[all …]
H A DExecutionPlan.h145 void declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex,
158 bool redeclare(SourceOperandIndex sourceOperandIndex, const Dimensions& newDimensions,
190 std::optional<LocationAndShape> lookup(SourceOperandIndex sourceOperandIndex,
242 int addOperand(uint32_t sourceOperandIndex, uint32_t* stepOperandIndex, OperandKind kind);
718 void recordOutputDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex);
719 void recordTemporaryDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex);
787 const Operand& getSourceOperand(const std::pair<uint32_t, uint32_t>& sourceOperandIndex) const { in getSourceOperand() argument
789 .getModel(sourceOperandIndex.first) in getSourceOperand()
790 ->getOperand(sourceOperandIndex.second); in getSourceOperand()
H A DExecutionBuilder.h112 const Operand& getSourceOperand(const std::pair<uint32_t, uint32_t>& sourceOperandIndex) const { in getSourceOperand() argument
113 return getSourceModel(sourceOperandIndex.first)->getOperand(sourceOperandIndex.second); in getSourceOperand()
H A DExecutionBuilder.cpp1240 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, it->second); in updateOutputShapes() local
1242 << " sourceOperandIndex = (" << sourceOperandIndex.first << ", " in updateOutputShapes()
1243 << sourceOperandIndex.second << ") is a dynamic temporary"; in updateOutputShapes()
1245 const auto loc = mDynamicTemporaries->lookup(sourceOperandIndex, false); in updateOutputShapes()
1254 changedShape = mDynamicTemporaries->redeclare(sourceOperandIndex, in updateOutputShapes()
1260 sourceOperandIndex, from[i].dimensions, 2 * loc->paddedLength); in updateOutputShapes()