Lines Matching refs:StepExecutor
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()
531 int StepExecutor::startCompute(sp<ExecutionCallback>* synchronizationCallback) { in startCompute()
543 int StepExecutor::startComputeOnDevice(sp<ExecutionCallback>* synchronizationCallback) { in startComputeOnDevice()
690 int StepExecutor::startComputeOnCpu(sp<ExecutionCallback>* synchronizationCallback) { in startComputeOnCpu()