Searched refs:asyncFree (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 172 cudaError_t err = cudaStreamAddCallback(*stream_, asyncFree, afData, 0); in deallocate()
175 hipError_t err = hipStreamAddCallback(*stream_, asyncFree, afData, 0); in deallocate()
202 static void CUDART_CB asyncFree(gpuStream_t stream, cudaError_t status, in asyncFree() function in tensorflow::EigenGpuStreamDevice
205 static void asyncFree(gpuStream_t stream, hipError_t status, void* userData) { in asyncFree()
|