Searched refs:scoped_activation (Results 1 – 10 of 10) sorted by relevance
42 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw() local
566 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in Compute() local630 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
262 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync() local265 se::rocm::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync()
312 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType() local
314 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
280 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
187 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
854 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid() local
233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync() local