| /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 | 66 cuda::ScopedActivateExecutorContext sac(parent); in SetStream()
87 cuda::ScopedActivateExecutorContext sac(parent); in Initialize()
258 cuda::ScopedActivateExecutorContext sac(parent_); in UpdateScratchAllocator()
269 cuda::ScopedActivateExecutorContext sac(parent_); in ~CUDAFftPlan()
496 cuda::ScopedActivateExecutorContext sac(parent_); in DoFftInternal()
525 cuda::ScopedActivateExecutorContext sac(parent_); in DoFftWithDirectionInternal()
|
| D | cuda_blas.cc | 223 gpu::ScopedActivateExecutorContext sac{parent_}; in Init()
253 gpu::ScopedActivateExecutorContext sac{parent_}; in ~CUDABlas()
258 gpu::ScopedActivateExecutorContext sac{parent_}; in ~CUDABlas()
268 gpu::ScopedActivateExecutorContext sac{parent_}; in SetStream()
281 gpu::ScopedActivateExecutorContext sac{parent_}; in CUDAStream()
556 gpu::ScopedActivateExecutorContext sac{parent_}; in DoBlasInternalImpl()
3475 gpu::ScopedActivateExecutorContext sac{parent_}; in DoBlasInternalImpl()
3618 gpu::ScopedActivateExecutorContext sac{parent_}; in DoBlasInternalImpl()
|
| D | cuda_dnn.cc | 136 CudnnHandle(gpu::ScopedActivateExecutorContext context, in CudnnHandle()
145 gpu::ScopedActivateExecutorContext context_;
186 gpu::ScopedActivateExecutorContext context(executor); in GetHandle()
310 ScopedActivateExecutorContext context(parent_); in Init()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_activation.h | 34 using ScopedActivateExecutorContext = gpu::ScopedActivateExecutorContext; variable
|
| D | rocm_rng.cc | 70 gpu::ScopedActivateExecutorContext sac{parent}; \
99 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| D | rocm_fft.cc | 52 gpu::ScopedActivateExecutorContext sac{parent}; \
81 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| D | rocm_dnn.cc | 193 MIOpenHandle(gpu::ScopedActivateExecutorContext context, in MIOpenHandle()
202 gpu::ScopedActivateExecutorContext context_;
573 gpu::ScopedActivateExecutorContext context(executor); in GetHandle()
610 ScopedActivateExecutorContext context(parent_); in Init()
|
| D | rocm_blas.cc | 59 gpu::ScopedActivateExecutorContext sac{parent}; \
92 gpu::ScopedActivateExecutorContext sac{parent}; \
|
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.cc | 44 using stream_executor::gpu::ScopedActivateExecutorContext;
54 ScopedActivateExecutorContext sac{parent}; \
84 ScopedActivateExecutorContext sac{parent}; \
|
| D | cuda_solvers.cc | 102 using se::cuda::ScopedActivateExecutorContext;
233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 42 using ::perftools::gputools::cuda::ScopedActivateExecutorContext;
45 using ::perftools::gputools::rocm::ScopedActivateExecutorContext;
187 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| D | dense_to_csr_sparse_matrix_op.cc | 44 using ::perftools::gputools::cuda::ScopedActivateExecutorContext;
47 using ::perftools::gputools::rocm::ScopedActivateExecutorContext;
233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op.cc | 45 using stream_executor::cuda::ScopedActivateExecutorContext;
48 using stream_executor::rocm::ScopedActivateExecutorContext;
312 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType()
|
| D | dynamic_partition_op_gpu.cu.cc | 53 using stream_executor::cuda::ScopedActivateExecutorContext;
56 using stream_executor::rocm::ScopedActivateExecutorContext;
314 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| D | check_numerics_op.cc | 262 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync()
265 se::rocm::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync()
|
| D | segment_reduction_ops_impl.h | 51 using stream_executor::cuda::ScopedActivateExecutorContext;
55 using stream_executor::rocm::ScopedActivateExecutorContext;
280 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| /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 | 92 using se::cuda::ScopedActivateExecutorContext;
99 using se::rocm::ScopedActivateExecutorContext;
566 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in Compute()
630 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_thunk.cc | 103 se::gpu::ScopedActivateExecutorContext scoped_context(executor); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 40 using se::cuda::ScopedActivateExecutorContext;
42 using se::rocm::ScopedActivateExecutorContext;
704 ScopedActivateExecutorContext scoped_context(nccl_stream->executor); in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 45 using stream_executor::cuda::ScopedActivateExecutorContext;
48 using stream_executor::rocm::ScopedActivateExecutorContext;
854 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid()
|