Searched defs:gpu_stream (Results 1 – 13 of 13) sorted by relevance
51 se::gpu::GpuStreamHandle gpu_stream = se::gpu::AsGpuStreamValue(&stream); in RunAllReduce() local379 se::gpu::GpuStreamHandle gpu_stream = se::gpu::AsGpuStreamValue(&stream); in RunReduceScatter() local
59 auto gpu_stream = se::gpu::AsGpuStreamValue(params.stream); in ExecuteOnStream() local
71 se::gpu::GpuStreamHandle gpu_stream = se::gpu::AsGpuStreamValue(&stream); in RunAllGather() local
87 se::gpu::GpuStreamHandle gpu_stream = se::gpu::AsGpuStreamValue(&stream); in RunAllToAll() local
206 se::gpu::GpuStreamHandle gpu_stream = se::gpu::AsGpuStreamValue(&stream); in RunCollectivePermute() local
302 const gpuStream_t* gpu_stream = in Create() local
128 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()903 const void* gpu_stream, in Reinitialize()916 const void* gpu_stream, in Reinitialize()1165 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local
70 GpuStreamHandle gpu_stream() const { in gpu_stream() function
302 Stream* FindAllocatedStream(void* gpu_stream) override { in FindAllocatedStream()
83 void* gpu_stream = nullptr; member
545 Stream* FindAllocatedStream(void* gpu_stream) { in FindAllocatedStream()
325 const void* gpu_stream = reinterpret_cast<const void*>( in ReinitializeGpuDevice() local
235 auto gpu_stream = device_plane_builder.GetOrCreateLine(0); in TEST() local