Searched refs:device_memory_size (Results 1 – 9 of 9) sorted by relevance
58 size_t device_memory_size = cuda_engine->getDeviceMemorySize(); in Create() local59 VLOG(2) << "Device memory size for cuda engine " << device_memory_size; in Create()61 if (device_memory_size > 0) { in Create()62 device_memory = allocator->allocate(device_memory_size, in Create()
85 port::HumanReadableNumBytes::ToString(device_memory_size()); in ToMap()
114 int64 device_memory_size() const { return device_memory_size_; } in device_memory_size() function
140 int64_t device_memory_size; member
369 builder.set_device_memory_size(description->device_memory_size); in CreateDeviceDescription()
879 uint64 device_memory_size = -1; in CreateDeviceDescription() local880 (void)GpuDriver::GetDeviceTotalMemory(device, &device_memory_size); in CreateDeviceDescription()881 builder.set_device_memory_size(device_memory_size); in CreateDeviceDescription()
1030 uint64 device_memory_size = -1; in CreateDeviceDescription() local1031 (void)GpuDriver::GetDeviceTotalMemory(device, &device_memory_size); in CreateDeviceDescription()1032 builder.set_device_memory_size(device_memory_size); in CreateDeviceDescription()
1778 description->device_memory_size()) in GetValidDeviceIds()1792 description->device_memory_size()) in GetValidDeviceIds()
2106 xla::int64 device_memory_size = 8LL * 1024 * 1024 * 1024; in TEST() local2108 device_memory_size = 16LL * 1024 * 1024 * 1024; in TEST()2158 xla::int64 num_tensors = 8 + device_memory_size / tensor_size; in TEST()