Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_thunk.cc59 auto gpu_stream = se::gpu::AsGpuStreamValue(params.stream); in ExecuteOnStream() local
Dcusolver_context.cc231 const gpuStream_t* gpu_stream = in Create() 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()
873 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
1095 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local