Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp277 << ", stepIndex = " << temp.second.stepIndex in vlogDump()
286 void DynamicTemporaries::declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex, in declare() argument
290 << toString(sourceOperandIndex) << ", stepIndex = " << stepIndex in declare()
298 sourceOperandIndex, InternalLocationAndShape{stepIndex, 0, initialDimensions, in declare()
301 mStepIndexToSourceOperandIndexes[stepIndex].emplace_back(sourceOperandIndex); in declare()
327 mAllocatedStepIndexes.erase(temp.stepIndex); in redeclare()
334 int DynamicTemporaries::allocate(uint32_t stepIndex) { in allocate() argument
335 VLOG(EXECUTION) << "DynamicTemporaries::allocate(stepIndex = " << stepIndex << ")"; in allocate()
339 const auto sourceOperandIndexesI = mStepIndexToSourceOperandIndexes.find(stepIndex); in allocate()
358 auto& memory = mStepIndexToMemory[stepIndex]; in allocate()
[all …]
DExecutionPlan.h145 void declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex,
169 int allocate(uint32_t stepIndex);
173 bool allocated(uint32_t stepIndex) const;
201 uint32_t stepIndex; member
238 ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex,
718 void recordOutputDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex);
719 void recordTemporaryDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex);