Searched refs:gpuStream_t (Results 1 – 1 of 1) sorted by relevance
87 typedef cudaStream_t gpuStream_t; typedef94 typedef hipStream_t gpuStream_t;120 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()138 const gpuStream_t& stream() const override { return *stream_; } in stream()200 static void CUDART_CB asyncFree(gpuStream_t stream, cudaError_t status, in asyncFree()203 static void asyncFree(gpuStream_t stream, hipError_t status, void* userData) { in asyncFree()216 const gpuStream_t* stream_; // Not owned. in asyncFree()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()