Searched defs:device_memory_size (Results 1 – 4 of 4) sorted by relevance
114 int64 device_memory_size() const { return device_memory_size_; } in device_memory_size() function
1069 uint64 device_memory_size = -1; in CreateDeviceDescription() local
904 uint64 device_memory_size = -1; in CreateDeviceDescription() local
1599 xla::int64 device_memory_size = 8LL * 1024 * 1024 * 1024; in TEST() local