Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.h99 typedef std::pair<uint32_t, uint32_t> SourceOperandIndex; typedef
145 void declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex,
158 bool redeclare(SourceOperandIndex sourceOperandIndex, const Dimensions& newDimensions,
190 std::optional<LocationAndShape> lookup(SourceOperandIndex sourceOperandIndex,
208 std::map<SourceOperandIndex, InternalLocationAndShape> mSourceOperandToTemporary;
211 std::map<uint32_t, std::vector<SourceOperandIndex>> mStepIndexToSourceOperandIndexes;
299 const std::map<SourceOperandIndex, StaticTemporaryLocation>&
302 const std::map<SourceOperandIndex, uint32_t>& sourceOperandToInputIndex,
303 const std::map<SourceOperandIndex, uint32_t>& sourceOperandToOutputIndex,
304 const std::map<SourceOperandIndex, ConstantReferenceLocation>&
[all …]
DExecutionPlan.cpp190 std::string toString(SourceOperandIndex sourceOperandIndex) { in toString()
213 static std::map<SourceOperandIndex, std::set<StepRole>> analyze( in analyze()
222 SourceOperandIndex source = {step.getSourceModelIndex(), operandIndex}; in addRole()
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;
286 void DynamicTemporaries::declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex, in declare()
[all …]
DExecutionBuilder.cpp1240 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, it->second); in updateOutputShapes()
/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp175 using SourceOperandIndex = ::android::nn::SourceOperandIndex; typedef
3376 static std::string toString(SourceOperandIndex index) { in toString()
3420 void checkStepRolesOfSourceOperand(SourceOperandIndex index, in checkStepRolesOfSourceOperand()