Home
last modified time | relevance | path

Searched refs:scoped_activation (Results 1 – 7 of 7) 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.cc568 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeHelper() local
634 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local
/external/tensorflow/tensorflow/core/kernels/
Dcheck_numerics_op.cc181 se::cuda::ScopedActivateExecutorContext scoped_activation{ in ComputeAsync() local
Dwhere_op.cc307 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsyncType() local
Dcrop_and_resize_op.cc812 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in RunIfBoxIndexIsValid() local
Dcuda_solvers.cc212 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in CheckLapackInfoAndDeleteSolverAsync() local
Dsegment_reduction_ops.cc269 ScopedActivateExecutorContext scoped_activation{stream->parent()}; in ComputeAsync() local