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