Lines Matching refs:StepExecutor
610 StepExecutor executor(executionBuilder, executionBuilder->getModel(), in cpuFallbackFull()
624 static std::tuple<int, std::vector<OutputShape>, Timing, std::shared_ptr<StepExecutor>>
631 std::shared_ptr<StepExecutor> executor; in cpuFallbackPartial()
695 std::shared_ptr<StepExecutor> executor; in computeInternal()
721 StepExecutor::UpdateOutputShapes updateOutputShapes = {}; in computeInternal()
777 StepExecutor::UpdateOutputShapes fallbackUpdateOutputShapes = {}; in computeInternal()
917 std::shared_ptr<StepExecutor> executor; in computeFencedInternal()
1184 std::string toString(StepExecutor::UpdateOutputShapes updateOutputShapes) { in toString()
1191 bool StepExecutor::updateOutputShapes(int executionResultCode, const std::vector<OutputShape>& from, in updateOutputShapes()
1313 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor() function in android::nn::StepExecutor
1333 bool StepExecutor::areDynamicTemporariesAllocated() const { in areDynamicTemporariesAllocated()
1337 void StepExecutor::mapInputsAndOutputsTrivially() { in mapInputsAndOutputsTrivially()
1343 void StepExecutor::mapInputOrOutput(const ModelArgumentInfo& builderInputOrOutput, in mapInputOrOutput()
1375 int StepExecutor::setInputOrOutputFromMemory(const Operand& inputOrOutputOperand, in setInputOrOutputFromMemory()
1439 bool StepExecutor::isCpu() const { in isCpu()
1443 std::pair<int, std::shared_ptr<RuntimeExecution>> StepExecutor::getReusableExecution() { in getReusableExecution()
1460 std::tuple<int, std::vector<OutputShape>, Timing> StepExecutor::compute( in compute()
1489 std::tuple<int, int, ExecuteFencedInfoCallback> StepExecutor::computeFenced( in computeFenced()
1529 std::tuple<int, std::vector<OutputShape>, Timing> StepExecutor::computeOnCpuFallback() { in computeOnCpuFallback()