Searched defs:SetStreamAndPreallocateMemory (Results 1 – 2 of 2) sorted by relevance
387 void GpuCudaMallocAsyncAllocator::SetStreamAndPreallocateMemory(void* stream) { in SetStreamAndPreallocateMemory() function in tensorflow::GpuCudaMallocAsyncAllocator
227 virtual void SetStreamAndPreallocateMemory(void* stream) {} in SetStreamAndPreallocateMemory() function