Searched refs:AsGpuStreamValue (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 301 GpuStreamHandle hipstream = AsGpuStreamValue(stream); in Launch()
520 AsGpuStreamValue(stream)); in Memset()
532 AsGpuStreamValue(stream)); in Memset32()
539 AsGpuStreamValue(stream)); in Memcpy()
546 AsGpuStreamValue(stream)); in Memcpy()
555 AsGpuStreamValue(stream)); in MemcpyDeviceToDevice()
566 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
630 AsGpuStreamValue(other)) in CreateStreamDependency()
638 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
651 return GpuDriver::SynchronizeStream(context_, AsGpuStreamValue(stream)); in BlockHostUntilDone()
|
| D | rocm_rng.cc | 144 wrap::hiprandSetStream(parent_, rng_, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_fft.cc | 147 auto ret = wrap::hipfftSetStream(parent, plan, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_blas.cc | 361 CHECK(AsGpuStreamValue(stream) != nullptr); in SetStream()
364 wrap::rocblas_set_stream(parent_, blas_, AsGpuStreamValue(stream)); in SetStream()
|
| D | rocm_dnn.cc | 574 hipStream_t hip_stream = stream ? AsGpuStreamValue(stream) : nullptr; in GetHandle()
4144 AsGpuStreamValue(stream)); in DoPoolForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 405 CUstream custream = AsGpuStreamValue(stream); in Launch()
612 AsGpuStreamValue(stream)); in Memset()
624 AsGpuStreamValue(stream)); in Memset32()
631 AsGpuStreamValue(stream)); in Memcpy()
638 AsGpuStreamValue(stream)); in Memcpy()
647 AsGpuStreamValue(stream)); in MemcpyDeviceToDevice()
658 return GpuDriver::AddStreamCallback(context_, AsGpuStreamValue(stream), in HostCallback()
722 AsGpuStreamValue(other)) in CreateStreamDependency()
730 return GpuDriver::WaitStreamOnEvent(context_, AsGpuStreamValue(dependent), in CreateStreamDependency()
743 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 | 67 auto ret = cufftSetStream(plan, AsGpuStreamValue(stream)); in SetStream()
|
| D | cuda_blas.cc | 266 CHECK(AsGpuStreamValue(stream) != nullptr); in SetStream()
269 cublasStatus_t ret = cublasSetStream(blas_, AsGpuStreamValue(stream)); in SetStream()
280 CHECK(AsGpuStreamValue(stream) != nullptr); in CUDAStream()
282 return AsGpuStreamValue(stream); in CUDAStream()
|
| D | cuda_dnn.cc | 187 CUstream cu_stream = stream ? AsGpuStreamValue(stream) : cudaStreamLegacy; in GetHandle()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.cc | 56 GpuStreamHandle AsGpuStreamValue(Stream* stream) { in AsGpuStreamValue() function
|
| D | gpu_stream.h | 94 GpuStreamHandle AsGpuStreamValue(Stream* stream);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_thunk.cc | 55 auto gpu_stream = se::gpu::AsGpuStreamValue(params.stream); in ExecuteOnStream()
|