Searched refs:ExecutionStep (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.h | 40 class ExecutionStep {
47 ExecutionStep(ExecutionPlan* plan,
202 std::shared_ptr<ExecutionStep> createNewStep(const std::shared_ptr<Device> device);
222 const std::vector<std::shared_ptr<ExecutionStep>>& forTest_compoundGetSteps() const;
260 std::vector<std::shared_ptr<ExecutionStep>> mSteps;
|
| D | ExecutionPlan.cpp | 131 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, in ExecutionStep() function in android::nn::ExecutionStep
137 int ExecutionStep::addOperand(uint32_t fromOperandIndex, uint32_t* toOperandIndex, in addOperand()
229 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation()
269 void ExecutionStep::mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const { in mapInputsAndOutputs()
291 void ExecutionStep::logSubModel() const { in logSubModel()
329 const ExecutionStep::RemapVectorType& myModelInputsOrOutputs, in convertModelInputsOrOutputs()
349 int ExecutionStep::finishSubModel(const ModelBuilder* fromModel, bool* hasOutputOfUnknownSize, in finishSubModel()
436 void ExecutionStep::dump() const { in dump()
700 std::shared_ptr<ExecutionStep> ExecutionPlan::createNewStep(const std::shared_ptr<Device> device) { in createNewStep()
707 auto step = std::make_shared<ExecutionStep>(this, steps.size(), device); in createNewStep()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestPartitioning.cpp | 130 using ExecutionStep = ::android::nn::ExecutionStep; typedef
429 using RemapVectorType = ExecutionStep::RemapVectorType;
430 using SubModelOutputSetType = ExecutionStep::SubModelOutputSetType;
743 bool compare(std::shared_ptr<const ExecutionStep> step, in compare()
|