Searched refs:allocator_memory (Results 1 – 1 of 1) sorted by relevance
79 size_t allocator_memory = free_memory * memory_fraction; in CreateCudaAsyncAllocator() local81 LOG(INFO) << "XLA backend allocating " << allocator_memory in CreateCudaAsyncAllocator()85 LOG(INFO) << "XLA backend will use up to " << allocator_memory in CreateCudaAsyncAllocator()91 tensorflow::PlatformDeviceId(device_ordinal), allocator_memory, in CreateCudaAsyncAllocator()240 size_t allocator_memory = enable_unified_memory in CreateBFCAllocator() local244 LOG(INFO) << "XLA backend allocating " << allocator_memory in CreateBFCAllocator()248 LOG(INFO) << "XLA backend will use up to " << allocator_memory in CreateBFCAllocator()256 std::move(sub_allocator), allocator_memory, in CreateBFCAllocator()