Home
last modified time | relevance | path

Searched refs:scoped_activation (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_cudamalloc_allocator.cc42 se::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_}; in AllocateRaw() local
Dgpu_device.cc566 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in Compute() local
630 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
/external/tensorflow/tensorflow/core/kernels/
Dcheck_numerics_op.cc262 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync() local
265 se::rocm::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync()
Dwhere_op.cc312 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType() local
Ddynamic_partition_op_gpu.cu.cc314 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
Dsegment_reduction_ops_impl.h280 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_tensor_to_csr_sparse_matrix_op.cc187 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
Ddense_to_csr_sparse_matrix_op.cc233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
/external/tensorflow/tensorflow/core/kernels/image/
Dcrop_and_resize_op.cc854 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid() local
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc233 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync() local