Searched refs:StepExecutor (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionBuilder.cpp | 223 StepExecutor executor(executionBuilder, executionBuilder->getModel(), in cpuFallbackFull()
248 std::shared_ptr<StepExecutor> executor; in cpuFallbackPartial()
274 std::shared_ptr<StepExecutor> executor; in asyncStartComputePartitioned()
398 StepExecutor executor(this, mModel, device->getInterface(), in startCompute()
409 StepExecutor executor(this, mModel, in startCompute()
418 int StepExecutor::allocatePointerArgumentsToPool(std::vector<ModelArgumentInfo>* args, in allocatePointerArgumentsToPool()
458 StepExecutor::StepExecutor(const ExecutionBuilder* executionBuilder, in StepExecutor() function in android::nn::StepExecutor
465 void StepExecutor::mapInputsAndOutputsTrivially() { in mapInputsAndOutputsTrivially()
471 void StepExecutor::mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput, in mapInputOrOutput()
492 int StepExecutor::setInputOrOutputFromTemporaryMemory(const Operand& inputOrOutputOperand, in setInputOrOutputFromTemporaryMemory()
[all …]
|
| D | ExecutionBuilder.h | 39 class StepExecutor; variable
67 friend class StepExecutor; variable
110 class StepExecutor {
121 StepExecutor(const ExecutionBuilder* executionBuilder,
|
| D | ExecutionPlan.h | 38 class StepExecutor; variable
93 void mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const;
197 int next(std::shared_ptr<Controller> controller, std::shared_ptr<StepExecutor>* executor) const;
200 …int fallback(std::shared_ptr<Controller> controller, std::shared_ptr<StepExecutor>* executor) cons…
|
| D | ExecutionPlan.cpp | 269 void ExecutionStep::mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const { in mapInputsAndOutputs()
562 std::shared_ptr<StepExecutor>* executor) const { in fallback()
583 std::shared_ptr<StepExecutor>* executor) const { in next()
604 *executor = std::make_shared<StepExecutor>( in next()
633 *executor = std::make_shared<StepExecutor>( in next()
|