Lines Matching refs:ExecutionStep
260 void addRole(const ExecutionStep& step, uint32_t operandIndex, IOType type, in addRole()
444 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex, in ExecutionStep() function in android::nn::ExecutionStep
455 int ExecutionStep::addOperand(uint32_t sourceOperandIndex, uint32_t* stepOperandIndex, in addOperand()
555 int ExecutionStep::addOperation(int operationIndex) { in addOperation()
589 void ExecutionStep::mapInputsAndOutputs( in mapInputsAndOutputs()
672 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in findModelOutputsThatAreDownstreamInputs()
695 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in findTempsAsStepModelOutputs()
715 void ExecutionStep::declareModelOutputIsDownstreamInput(uint32_t mainModelOutputIndex) { in declareModelOutputIsDownstreamInput()
726 void ExecutionStep::recordTempAsStepModelOutput(uint32_t stepOperandIndex) { in recordTempAsStepModelOutput()
732 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
736 void ExecutionStep::logStepModel() const { in logStepModel()
784 int ExecutionStep::finishStepModel(const ModelBuilder* mainModel, bool* hasOutputOfUnknownSize, in finishStepModel()
879 void ExecutionStep::dump() const { in dump()
938 if (ExecutionStep* step = logicalStep->tryExecutionStep()) { in finish()
1053 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in findMemoryStepRoles()
1330 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in makeController()
1601 } else if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in nextCompound()
1609 int ExecutionPlan::nextCompound(const ExecutionStep* step, std::shared_ptr<Controller> controller, in nextCompound()
1905 ExecutionStep* ExecutionPlan::createNewExecutionStep(uint32_t sourceModelIndex, in createNewExecutionStep()
1908 auto step = std::make_shared<LogicalStep>(std::in_place_type<ExecutionStep>, this, in createNewExecutionStep()
2022 const ExecutionStep* step = logicalStep->tryExecutionStep(); in hasStepModelWithNoInputsOrNoOutputs()
2121 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in forEachDynamicTemporary()
2255 ExecutionStep* step = in partitionTheWorkInternal()