Home
last modified time | relevance | path

Searched refs:ExecutionBuilder (Results 1 – 8 of 8) sorted by relevance

/frameworks/ml/nn/runtime/
DCompilationBuilder.h30 class ExecutionBuilder; variable
35 friend class ExecutionBuilder; // TODO remove this
47 int createExecution(ExecutionBuilder** execution);
DNeuralNetworks.cpp416 ExecutionBuilder* r = nullptr; in ANeuralNetworksExecution_create()
425 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_free()
436 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_setInput()
450 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_setInputFromMemory()
461 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_setOutput()
474 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_setOutputFromMemory()
487 ExecutionBuilder* r = reinterpret_cast<ExecutionBuilder*>(execution); in ANeuralNetworksExecution_startCompute()
DExecutionBuilder.h66 class ExecutionBuilder {
69 ExecutionBuilder(const CompilationBuilder* compilation);
121 StepExecutor(const ExecutionBuilder* executionBuilder,
180 const ExecutionBuilder* mExecutionBuilder;
DExecutionBuilder.cpp128 ExecutionBuilder::ExecutionBuilder(const CompilationBuilder* compilation) : in ExecutionBuilder() function in android::nn::ExecutionBuilder
137 int ExecutionBuilder::setInput(uint32_t index, const ANeuralNetworksOperandType* type, in setInput()
159 int ExecutionBuilder::setInputFromMemory(uint32_t index, const ANeuralNetworksOperandType* type, in setInputFromMemory()
178 int ExecutionBuilder::setOutput(uint32_t index, const ANeuralNetworksOperandType* type, void* buffe… in setOutput()
199 int ExecutionBuilder::setOutputFromMemory(uint32_t index, const ANeuralNetworksOperandType* type, in setOutputFromMemory()
220 static void cpuFallbackFull(const ExecutionBuilder* executionBuilder, in cpuFallbackFull()
243 static bool cpuFallbackPartial(const ExecutionBuilder* executionBuilder, in cpuFallbackPartial()
267 static void asyncStartComputePartitioned(const ExecutionBuilder* executionBuilder, in asyncStartComputePartitioned()
327 int ExecutionBuilder::startCompute(sp<ExecutionCallback>* synchronizationCallback) { in startCompute()
458 StepExecutor::StepExecutor(const ExecutionBuilder* executionBuilder, in StepExecutor()
DExecutionPlan.h35 class ExecutionBuilder; variable
184 Controller(const ExecutionPlan* plan, const ExecutionBuilder* executionBuilder,
189 const ExecutionBuilder* mExecutionBuilder;
195 std::shared_ptr<Controller> makeController(const ExecutionBuilder* executionBuilder) const;
DCompilationBuilder.cpp100 int CompilationBuilder::createExecution(ExecutionBuilder **execution) { in createExecution()
106 *execution = new (std::nothrow) ExecutionBuilder(this); in createExecution()
DAndroid.bp39 "ExecutionBuilder.cpp",
DExecutionPlan.cpp486 const ExecutionBuilder* executionBuilder, in Controller()
500 const ExecutionBuilder* executionBuilder) const { in makeController()