Searched refs:ExecutionStep (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionPlan.h | 231 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,
|
| D | ExecutionPlan.cpp | 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()
[all …]
|
| D | ExecutionBuilder.h | 49 class ExecutionStep; variable
347 const ExecutionStep* step = nullptr,
442 const ExecutionStep* mExecutionStep;
|
| D | ExecutionBuilder.cpp | 1363 const ExecutionStep* step, DynamicTemporaries* dynamicTemporaries) in StepExecutor()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioningRandom.cpp | 107 using ExecutionStep = nn::ExecutionStep; typedef
|
| D | TestPartitioning.cpp | 163 using ExecutionStep = ::android::nn::ExecutionStep; typedef
997 using RemapVectorType = ExecutionStep::RemapVectorType;
998 using StepModelOutputSetType = ExecutionStep::StepModelOutputSetType;
1528 bool compare(const ExecutionStep* step, const PartitioningModel* model, in compare()
1544 const ExecutionStep* step = logicalStep->executionStep(); in compare()
|