| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_activation.cc | 28 ScopedActivateExecutorContext::ScopedActivateExecutorContext( in ScopedActivateExecutorContext() function in stream_executor::gpu::ScopedActivateExecutorContext
33 ScopedActivateExecutorContext::ScopedActivateExecutorContext( in ScopedActivateExecutorContext() function in stream_executor::gpu::ScopedActivateExecutorContext
35 : ScopedActivateExecutorContext(ExtractGpuExecutor(stream_exec)) {} in ScopedActivateExecutorContext()
37 ScopedActivateExecutorContext::~ScopedActivateExecutorContext() { in ~ScopedActivateExecutorContext()
41 ScopedActivateExecutorContext::ScopedActivateExecutorContext( in ScopedActivateExecutorContext() function in stream_executor::gpu::ScopedActivateExecutorContext
42 ScopedActivateExecutorContext&& other) in ScopedActivateExecutorContext()
|
| D | gpu_activation.h | 38 class ScopedActivateExecutorContext {
41 explicit ScopedActivateExecutorContext(GpuExecutor* gpu_exec);
45 explicit ScopedActivateExecutorContext(StreamExecutor* stream_exec);
47 ScopedActivateExecutorContext(ScopedActivateExecutorContext&& other);
49 ~ScopedActivateExecutorContext();
55 SE_DISALLOW_COPY_AND_ASSIGN(ScopedActivateExecutorContext);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_rng.cc | 68 cuda::ScopedActivateExecutorContext sac(parent_); in ~GpuRng()
77 cuda::ScopedActivateExecutorContext sac(parent_); in Init()
89 cuda::ScopedActivateExecutorContext sac(parent_); in SetStream()
124 cuda::ScopedActivateExecutorContext sac(parent_); in DoPopulateRandUniformInternal()
173 cuda::ScopedActivateExecutorContext sac(parent_); in DoPopulateRandGaussianInternal()
211 cuda::ScopedActivateExecutorContext sac(parent_); in SetSeed()
|
| D | cuda_activation.h | 34 using ScopedActivateExecutorContext = gpu::ScopedActivateExecutorContext; variable
|
| D | cuda_fft.cc | 65 cuda::ScopedActivateExecutorContext sac(parent); in SetStream()
85 cuda::ScopedActivateExecutorContext sac(parent); in Initialize()
255 cuda::ScopedActivateExecutorContext sac(parent_); in UpdateScratchAllocator()
266 cuda::ScopedActivateExecutorContext sac(parent_); in ~CUDAFftPlan()
468 cuda::ScopedActivateExecutorContext sac(parent_); in DoFftInternal()
496 cuda::ScopedActivateExecutorContext sac(parent_); in DoFftWithDirectionInternal()
|
| D | cuda_dnn.cc | 137 CudnnHandle(gpu::ScopedActivateExecutorContext context, mutex_lock lock, in CudnnHandle()
146 gpu::ScopedActivateExecutorContext context_;
186 gpu::ScopedActivateExecutorContext context(executor); in GetHandle()
301 ScopedActivateExecutorContext context(parent_); in Init()
|
| D | cuda_blas.cc | 235 gpu::ScopedActivateExecutorContext sac{parent_}; in Init()
250 gpu::ScopedActivateExecutorContext sac{parent_}; in ~CUDABlas()
259 gpu::ScopedActivateExecutorContext sac{parent_}; in SetStream()
412 gpu::ScopedActivateExecutorContext sac{parent_}; in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_activation.h | 34 using ScopedActivateExecutorContext = gpu::ScopedActivateExecutorContext; variable
|
| D | rocm_rng.cc | 71 gpu::ScopedActivateExecutorContext sac{parent}; \
100 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| D | rocm_fft.cc | 52 gpu::ScopedActivateExecutorContext sac{parent}; \
81 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| D | rocm_dnn.cc | 101 MIOpenHandle(gpu::ScopedActivateExecutorContext context, mutex_lock lock, in MIOpenHandle()
110 gpu::ScopedActivateExecutorContext context_;
472 gpu::ScopedActivateExecutorContext context(executor); in GetHandle()
490 ScopedActivateExecutorContext context(parent_); in Init()
|
| D | rocm_blas.cc | 59 gpu::ScopedActivateExecutorContext sac{parent}; \
92 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_cudamalloc_allocator.cc | 42 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw()
|
| D | gpu_device.cc | 90 using se::cuda::ScopedActivateExecutorContext;
97 using se::rocm::ScopedActivateExecutorContext;
568 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeHelper()
634 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op.cc | 45 using stream_executor::cuda::ScopedActivateExecutorContext;
307 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType()
|
| D | check_numerics_op.cc | 181 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync()
|
| D | crop_and_resize_op.cc | 42 using stream_executor::cuda::ScopedActivateExecutorContext;
812 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid()
|
| D | cuda_solvers.cc | 81 using se::cuda::ScopedActivateExecutorContext;
212 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync()
|
| D | segment_reduction_ops.cc | 45 using stream_executor::cuda::ScopedActivateExecutorContext;
269 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 43 using se::cuda::ScopedActivateExecutorContext;
543 ScopedActivateExecutorContext scoped_context(nccl_stream->executor); in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.cc | 285 se::cuda::ScopedActivateExecutorContext scoped_context(executor); in DoAllReduce()
|