Searched refs:burstController (Results 1 – 4 of 4) sorted by relevance
457 std::shared_ptr<ExecutionBurstController> burstController = nullptr; in asyncStartComputePartitioned() local458 int n = plan->next(controller, &executor, &burstController); in asyncStartComputePartitioned()473 n = executor->startCompute(&stepCallback, burstController); in asyncStartComputePartitioned()799 const std::shared_ptr<ExecutionBurstController>& burstController) { in startCompute() argument807 return startComputeOnDevice(synchronizationCallback, burstController); in startCompute()813 const std::shared_ptr<ExecutionBurstController>& burstController) { in startComputeOnDevice() argument902 const bool burstCompute = (burstController != nullptr); in startComputeOnDevice()908 memory->usedBy(burstController); in startComputeOnDevice()915 burstController->tryCompute(request, measureTiming(mExecutionBuilder), memoryIds); in startComputeOnDevice()
220 const std::shared_ptr<ExecutionBurstController>& burstController = nullptr);236 const std::shared_ptr<ExecutionBurstController>& burstController);
762 std::shared_ptr<ExecutionBurstController>* burstController) const { in next()764 if (burstController != nullptr) { in next()765 *burstController = nullptr; in next()790 if (burstController != nullptr && controller->mBurstBuilder != nullptr) { in next()791 *burstController = controller->mBurstBuilder->getControllerAt(0); in next()820 if (burstController != nullptr && controller->mBurstBuilder != nullptr) { in next()821 *burstController = controller->mBurstBuilder->getControllerAt(controller->mNextStepIndex); in next()
219 std::shared_ptr<ExecutionBurstController>* burstController = nullptr) const;