Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp12 namespace c10::cuda::CUDACachingAllocator::CudaMallocAsync { namespace
H A DCUDACachingAllocator.cpp3739 namespace CudaMallocAsync { namespace
3769 return CudaMallocAsync::allocator(); in parseEnvForBackend()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device_test.cc127 TEST_F(GPUDeviceTest, DISABLED_ON_GPU_ROCM(CudaMallocAsync)) { in TEST_F() argument