Home
last modified time | relevance | path

Searched refs:executionBuilder (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp382 static void cpuFallbackFull(ExecutionBuilder* executionBuilder, in cpuFallbackFull() argument
386 StepExecutor executor(executionBuilder, executionBuilder->getModel(), in cpuFallbackFull()
410 static bool cpuFallbackPartial(ExecutionBuilder* executionBuilder, const ExecutionPlan* plan, in cpuFallbackPartial() argument
419 cpuFallbackFull(executionBuilder, executionCallback); in cpuFallbackPartial()
424 cpuFallbackFull(executionBuilder, executionCallback); in cpuFallbackPartial()
438 cpuFallbackFull(executionBuilder, executionCallback); in cpuFallbackPartial()
445 static void asyncStartComputePartitioned(ExecutionBuilder* executionBuilder, in asyncStartComputePartitioned() argument
453 executionBuilder->initializeOutputShapes(&outputShapes); in asyncStartComputePartitioned()
461 cpuFallbackFull(executionBuilder, executionCallback); in asyncStartComputePartitioned()
476 if (cpuFallbackPartial(executionBuilder, plan, controller, executionCallback, in asyncStartComputePartitioned()
[all …]
DExecutionPlan.h200 Controller(const ExecutionPlan* plan, ExecutionBuilder* executionBuilder,
215 std::shared_ptr<Controller> makeController(ExecutionBuilder* executionBuilder,
DExecutionBuilder.h179 StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model,
DExecutionPlan.cpp627 const ExecutionPlan* plan, ExecutionBuilder* executionBuilder, in Controller() argument
632 mExecutionBuilder(executionBuilder), in Controller()
682 ExecutionBuilder* executionBuilder, const BurstBuilder* burstBuilder) const { in makeController() argument
709 const ModelBuilder* fromModel = executionBuilder->getModel(); in makeController()
732 return std::shared_ptr<Controller>(new Controller(this, executionBuilder, burstBuilder, in makeController()