Searched defs:device_memory_size (Results 1 – 7 of 7) sorted by relevance
45 uint64 device_memory_size = -1; in DeviceIdToIdentifierHelper() local
151 int64 device_memory_size() const { return device_memory_size_; } in device_memory_size() function
87 const size_t device_memory_size = in AllocateDeviceMemory() local
1027 uint64 device_memory_size = -1; in CreateDeviceDescription() local
875 uint64 device_memory_size = -1; in CreateDeviceDescription() local
140 int64_t device_memory_size; member
2106 xla::int64 device_memory_size = 8LL * 1024 * 1024 * 1024; in TEST() local