Home
last modified time | relevance | path

Searched refs:ExecutionStep (Results 1 – 6 of 6) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionPlan.h231 class ExecutionStep {
238 ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex,
498 bool isExecution() const { return std::holds_alternative<ExecutionStep>(mStep); } in isExecution()
504 ExecutionStep* executionStep() { return &std::get<ExecutionStep>(mStep); } in executionStep()
510 const ExecutionStep* executionStep() const { return &std::get<ExecutionStep>(mStep); } in executionStep()
516 ExecutionStep* tryExecutionStep() { return std::get_if<ExecutionStep>(&mStep); } in tryExecutionStep()
522 const ExecutionStep* tryExecutionStep() const { return std::get_if<ExecutionStep>(&mStep); } in tryExecutionStep()
530 std::variant<ExecutionStep, IfStep, WhileStep, GotoStep> mStep;
703 ExecutionStep* createNewExecutionStep(uint32_t sourceModelIndex,
819 int nextCompound(const ExecutionStep* step, std::shared_ptr<Controller> controller,
H A DExecutionPlan.cpp404 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex, in ExecutionStep() function in android::nn::ExecutionStep
515 int ExecutionStep::addOperation(int operationIndex) { in addOperation()
549 void ExecutionStep::mapInputsAndOutputs( in mapInputsAndOutputs()
632 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in findModelOutputsThatAreDownstreamInputs()
655 if (const ExecutionStep* step = logicalStep->tryExecutionStep()) { in findTempsAsStepModelOutputs()
692 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
696 void ExecutionStep::logStepModel() const { in logStepModel()
839 void ExecutionStep::dump() const { in dump()
895 if (ExecutionStep* step = logicalStep->tryExecutionStep()) { in finish()
1970 const ExecutionStep* step = logicalStep->tryExecutionStep(); in hasStepModelWithNoInputsOrNoOutputs()
[all …]
H A DExecutionBuilder.h48 class ExecutionStep; variable
325 const ExecutionStep* step = nullptr,
420 const ExecutionStep* mExecutionStep;
H A DExecutionBuilder.cpp1316 const ExecutionStep* step, DynamicTemporaries* dynamicTemporaries) in StepExecutor()
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestPartitioningRandom.cpp110 using ExecutionStep = nn::ExecutionStep; typedef
H A DTestPartitioning.cpp162 using ExecutionStep = ::android::nn::ExecutionStep; typedef
996 using RemapVectorType = ExecutionStep::RemapVectorType;
997 using StepModelOutputSetType = ExecutionStep::StepModelOutputSetType;
1527 bool compare(const ExecutionStep* step, const PartitioningModel* model, in compare()
1543 const ExecutionStep* step = logicalStep->executionStep(); in compare()