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 | 220 void addRole(const ExecutionStep& step, uint32_t operandIndex, IOType type, in addRole()
404 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex, in ExecutionStep() function in android::nn::ExecutionStep
415 int ExecutionStep::addOperand(uint32_t sourceOperandIndex, uint32_t* stepOperandIndex, in addOperand()
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()
675 void ExecutionStep::declareModelOutputIsDownstreamInput(uint32_t mainModelOutputIndex) { in declareModelOutputIsDownstreamInput()
686 void ExecutionStep::recordTempAsStepModelOutput(uint32_t stepOperandIndex) { in recordTempAsStepModelOutput()
692 const ModelBuilder* ExecutionStep::getSourceModel() const { in getSourceModel()
[all …]
|
| D | ExecutionBuilder.h | 48 class ExecutionStep; variable
325 const ExecutionStep* step = nullptr,
420 const ExecutionStep* mExecutionStep;
|
| D | ExecutionBuilder.cpp | 1316 const ExecutionStep* step, DynamicTemporaries* dynamicTemporaries) in StepExecutor()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioning.cpp | 162 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()
|
| D | TestPartitioningRandom.cpp | 110 using ExecutionStep = nn::ExecutionStep; typedef
|