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.h99 typedef std::pair<uint32_t, uint32_t> SourceOperandIndex; typedef
424 std::vector<SourceOperandIndex> outerInputOperands;
426 std::vector<SourceOperandIndex> outerOutputOperands;
428 std::vector<SourceOperandIndex> thenBranchInputOperands;
430 std::vector<SourceOperandIndex> thenBranchOutputOperands;
432 std::vector<SourceOperandIndex> elseBranchInputOperands;
461 std::vector<SourceOperandIndex> outerInputOperands;
463 std::vector<SourceOperandIndex> outerOutputOperands;
465 std::vector<SourceOperandIndex> condInputOperands;
470 std::vector<SourceOperandIndex> bodyInputOperands;
[all …]
H A DExecutionPlan.cpp190 std::string toString(SourceOperandIndex sourceOperandIndex) { in toString()
213 static std::map<SourceOperandIndex, std::set<StepRole>> analyze( in analyze()
226 void setUsedBy(const SourceOperandIndex& source, const SourceOperandIndex& dest) { in setUsedBy()
235 std::map<SourceOperandIndex, std::set<StepRole>> finish() { in finish()
242 void finishHelper(SourceOperandIndex current) { in finishHelper()
258 std::map<SourceOperandIndex, std::set<StepRole>> mRoles;
261 std::map<SourceOperandIndex, std::set<SourceOperandIndex>> mUsedBy;
263 std::set<SourceOperandIndex> mProcessedOperands;
617 [this](const SourceOperandIndex& sourceOperandIndex) { in findModelOutputsThatAreDownstreamInputs()
1672 const std::vector<SourceOperandIndex>& branchInputOperands = in nextCompound()
[all …]
H A DExecutionBuilder.cpp1240 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, it->second); in updateOutputShapes()
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestPartitioning.cpp175 using SourceOperandIndex = ::android::nn::SourceOperandIndex; typedef
3376 static std::string toString(SourceOperandIndex index) { in toString()
3420 void checkStepRolesOfSourceOperand(SourceOperandIndex index, in checkStepRolesOfSourceOperand()