Home
last modified time | relevance | path

Searched refs:gpuStream_t (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc87 typedef cudaStream_t gpuStream_t; typedef
94 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()