HomeSort by relevance Sort by last modified time
    Searched defs:GPUBFCAllocator (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_bfc_allocator.cc 25 GPUBFCAllocator::GPUBFCAllocator(CudaGpuId cuda_gpu_id, size_t total_memory,
27 : GPUBFCAllocator(cuda_gpu_id, total_memory, GPUOptions(), name) {}
29 GPUBFCAllocator::GPUBFCAllocator(CudaGpuId cuda_gpu_id, size_t total_memory,
gpu_bfc_allocator.h 38 class GPUBFCAllocator : public BFCAllocator {
42 GPUBFCAllocator(CudaGpuId cuda_gpu_id, size_t total_memory,
44 GPUBFCAllocator(CudaGpuId cuda_gpu_id, size_t total_memory,
46 virtual ~GPUBFCAllocator() {}
48 TF_DISALLOW_COPY_AND_ASSIGN(GPUBFCAllocator);

Completed in 60 milliseconds