Lines Matching refs:stepIndex

317                         << ", stepIndex = " << temp.second.stepIndex  in vlogDump()
326 void DynamicTemporaries::declare(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex, in declare() argument
330 << toString(sourceOperandIndex) << ", stepIndex = " << stepIndex in declare()
338 sourceOperandIndex, InternalLocationAndShape{stepIndex, 0, initialDimensions, in declare()
341 mStepIndexToSourceOperandIndexes[stepIndex].emplace_back(sourceOperandIndex); in declare()
367 mAllocatedStepIndexes.erase(temp.stepIndex); in redeclare()
374 int DynamicTemporaries::allocate(uint32_t stepIndex) { in allocate() argument
375 VLOG(EXECUTION) << "DynamicTemporaries::allocate(stepIndex = " << stepIndex << ")"; in allocate()
379 const auto sourceOperandIndexesI = mStepIndexToSourceOperandIndexes.find(stepIndex); in allocate()
398 auto& memory = mStepIndexToMemory[stepIndex]; in allocate()
407 << " for step " << stepIndex; in allocate()
408 mAllocatedStepIndexes.erase(stepIndex); in allocate()
413 mAllocatedStepIndexes.insert(stepIndex); in allocate()
417 bool DynamicTemporaries::allocated(uint32_t stepIndex) const { in allocated()
418 return (mStepIndexToSourceOperandIndexes.find(stepIndex) == in allocated()
420 mAllocatedStepIndexes.count(stepIndex); in allocated()
429 const bool isAllocated = allocated(temp.stepIndex); in lookup()
435 return LocationAndShape{mStepIndexToMemory.at(temp.stepIndex).get(), temp.offset, in lookup()
444 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex, in ExecutionStep() argument
447 mIndex(stepIndex), in ExecutionStep()
660 uint32_t stepIndex = it->second; in findModelOutputsThatAreDownstreamInputs() local
661 CHECK_LT(stepIndex, mSteps.size()); in findModelOutputsThatAreDownstreamInputs()
663 << "ExecutionStep(" << stepIndex in findModelOutputsThatAreDownstreamInputs()
668 mSteps[stepIndex]->executionStep()->declareModelOutputIsDownstreamInput( in findModelOutputsThatAreDownstreamInputs()
690 uint32_t stepIndex = it->second; in findTempsAsStepModelOutputs() local
691 CHECK_LT(stepIndex, mSteps.size()); in findTempsAsStepModelOutputs()
692 mSteps[stepIndex]->executionStep()->recordTempAsStepModelOutput(sourceOperandIndex.second); in findTempsAsStepModelOutputs()
1056 const auto& [sourceIndex, stepIndex] = stepModelInputs[i]; in findMemoryStepRoles()
1061 const auto& [sourceIndex, stepIndex] = stepModelOutputs[i]; in findMemoryStepRoles()
1945 void ExecutionPlan::recordOutputDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex) { in recordOutputDef() argument
1947 compound()->mOutputToDefiningExecutionStep.emplace(sourceOperandIndex, stepIndex); in recordOutputDef()
1948 CHECK(isNew) << "Step " << stepIndex << " redefines output operand " in recordOutputDef()
1952 void ExecutionPlan::recordTemporaryDef(SourceOperandIndex sourceOperandIndex, uint32_t stepIndex) { in recordTemporaryDef() argument
1954 compound()->mTemporaryToDefiningExecutionStep.emplace(sourceOperandIndex, stepIndex); in recordTemporaryDef()
1955 CHECK(isNew) << "Step " << stepIndex << " redefines temporary operand " in recordTemporaryDef()
2083 for (const auto& [stepIndex, type, ioIndex] : it->second) { in forEachStepRoleOfSourceOperand()
2084 CHECK_LT(stepIndex, mSteps.size()); in forEachStepRoleOfSourceOperand()
2085 const auto* step = mSteps[stepIndex]->executionStep(); in forEachStepRoleOfSourceOperand()
2122 const uint32_t stepIndex = step->getIndex(); in forEachDynamicTemporary() local
2128 fn(sourceOperandIndex, sourceOperand, stepIndex); in forEachDynamicTemporary()