Home
last modified time | relevance | path

Searched defs:gpu_stream (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_thunk.cc55 auto gpu_stream = se::gpu::AsGpuStreamValue(params.stream); in ExecuteOnStream() local
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_stream.h70 GpuStreamHandle gpu_stream() const { in gpu_stream() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc122 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
765 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
1038 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local