Searched refs:AsGpuStreamValue (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 290 GpuStreamHandle hipstream = AsGpuStreamValue(stream); in Launch()
521 AsGpuStreamValue(stream)); in Memset()
533 AsGpuStreamValue(stream)); in Memset32()
540 AsGpuStreamValue(stream)); in Memcpy()
547 AsGpuStreamValue(stream)); in Memcpy()
556 AsGpuStreamValue(stream)); in MemcpyDeviceToDevice()
567 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
631 AsGpuStreamValue(other)) in CreateStreamDependency()
639 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
652 return GpuDriver::SynchronizeStream(context_, AsGpuStreamValue(stream)); in BlockHostUntilDone()
|
| D | rocm_rng.cc | 145 wrap::hiprandSetStream(parent_, rng_, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_fft.cc | 135 auto ret = wrap::hipfftSetStream(parent, plan, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_blas.cc | 317 CHECK(AsGpuStreamValue(stream) != nullptr); in SetStream()
320 wrap::rocblas_set_stream(parent_, blas_, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_dnn.cc | 473 hipStream_t hip_stream = stream ? AsGpuStreamValue(stream) : nullptr; in GetHandle()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 424 CUstream custream = AsGpuStreamValue(stream); in Launch()
640 AsGpuStreamValue(stream)); in Memset()
652 AsGpuStreamValue(stream)); in Memset32()
659 AsGpuStreamValue(stream)); in Memcpy()
666 AsGpuStreamValue(stream)); in Memcpy()
675 AsGpuStreamValue(stream)); in MemcpyDeviceToDevice()
686 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
750 AsGpuStreamValue(other)) in CreateStreamDependency()
758 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
771 return GpuDriver::SynchronizeStream(context_, AsGpuStreamValue(stream)); in BlockHostUntilDone()
|
| D | cuda_rng.cc | 90 curandStatus_t ret = curandSetStream(rng_, AsGpuStreamValue(stream)); in SetStream()
|
| D | cuda_fft.cc | 66 auto ret = cufftSetStream(plan, AsGpuStreamValue(stream)); in SetStream()
|
| D | cuda_blas.cc | 257 CHECK(AsGpuStreamValue(stream) != nullptr); in SetStream()
260 cublasStatus_t ret = cublasSetStream(blas_, AsGpuStreamValue(stream)); in SetStream()
|
| D | cuda_dnn.cc | 187 CUstream cu_stream = stream ? AsGpuStreamValue(stream) : cudaStreamLegacy; in GetHandle()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.cc | 55 GpuStreamHandle AsGpuStreamValue(Stream* stream) { in AsGpuStreamValue() function
|
| D | gpu_stream.h | 91 GpuStreamHandle AsGpuStreamValue(Stream* stream);
|