Searched defs:device_string (Results 1 – 9 of 9) sorted by relevance
131 std::string device_string = GetDeviceString(params.nccl_params); in RunNcclCollective() local144 absl::string_view device_string, int64_t current_id) { in RunCollectivePermute()
2037 std::string device_string = NcclCollectiveThunk::GetDeviceString(params); in operator ()() local
54 string device_string = "CPU"; in PersistentMemoryTrackingTest() local
91 bool IsPropagatableDevice(StringPiece device_string) { in IsPropagatableDevice()
64 std::string device_string; in GetDeviceNumber() local
53 def set_cpu0(device_string): argument
1066 def check_device(path, device_string): argument
588 char device_string[25]; in validate_device() local
1368 bool IsTpuDevice(StringPiece device_string) { in IsTpuDevice()