Searched defs:gpu_stream (Results 1 – 2 of 2) sorted by relevance
68 GpuStreamHandle gpu_stream() const { in gpu_stream() function
120 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()753 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()936 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local