Searched refs:ExecutionStep (Results 1 - 3 of 3) sorted by relevance

/frameworks/ml/nn/runtime/
H A DExecutionPlan.h40 class ExecutionStep { class in namespace:android::nn
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;
H A DExecutionPlan.cpp52 LOG(ERROR) << "ExecutionStep::finishSubModel compilation failed due to transport error: "
57 LOG(ERROR) << "ExecutionStep::finishSubModel compilation failed with error: "
131 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, function in class:android::nn::ExecutionStep
137 int ExecutionStep::addOperand(uint32_t fromOperandIndex, uint32_t* toOperandIndex,
229 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) {
269 void ExecutionStep::mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const {
291 void ExecutionStep::logSubModel() const {
292 VLOG(COMPILATION) << "ExecutionStep::finishSubModel, step " << mIndex;
329 const ExecutionStep
[all...]
/frameworks/ml/nn/runtime/test/
H A DTestPartitioning.cpp100 // (see ExecutionStep methods getModelInputs(), getModelOutputs(),
103 // operand number in the original model (see ExecutionStep method
130 using ExecutionStep = ::android::nn::ExecutionStep;
429 using RemapVectorType = ExecutionStep::RemapVectorType;
430 using SubModelOutputSetType = ExecutionStep::SubModelOutputSetType;
743 bool compare(std::shared_ptr<const ExecutionStep> step,

Completed in 37 milliseconds