Home
last modified time | relevance | path

Searched defs:scoped_activation (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_cudamallocasync_allocator.cc139 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in GpuCudaMallocAsyncAllocator() local
281 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw() local
333 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in DeallocateRaw() local
H A Dgpu_cudamalloc_allocator.cc39 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw() local
H A Dgpu_device.cc674 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in Compute() local
768 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_slice_op_gpu.cu.cc213 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in operator ()() local
H A Dsparse_to_dense_op_gpu.cu.cc195 se::gpu::ScopedActivateExecutorContext scoped_activation{ in operator ()() local
H A Dwhere_op.cc297 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType() local
H A Dcheck_numerics_op.cc262 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync() local
H A Dsparse_split_op_gpu.cu.cc311 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in operator ()() local
H A Ddynamic_partition_op_gpu.cu.cc303 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
H A Dsparse_fill_empty_rows_op_gpu.cu.cc364 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc203 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
H A Ddense_to_csr_sparse_matrix_op.cc233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Drocm_solvers.cc187 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync() local
H A Dcuda_solvers.cc232 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dcrop_and_resize_op.cc881 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid() local