Searched refs:computation_devices (Results 1 – 6 of 6) sorted by relevance
67 const auto& computation_devices = in GenerateXlaDeviceAssignment() local68 xrt_device_assignment.computation_devices(c); in GenerateXlaDeviceAssignment()69 if (num_replicas != computation_devices.replica_devices_size()) { in GenerateXlaDeviceAssignment()74 " replica_devices=", computation_devices.replica_devices_size()); in GenerateXlaDeviceAssignment()76 for (int64_t r = 0; r < computation_devices.replica_devices_size(); ++r) { in GenerateXlaDeviceAssignment()77 const auto& coords = computation_devices.replica_devices(r); in GenerateXlaDeviceAssignment()
481 xla_device_assignment->computation_devices(0); in TEST()484 xla_device_assignment->computation_devices(1); in TEST()587 xla_device_assignment->computation_devices(0); in TEST()590 xla_device_assignment->computation_devices(1); in TEST()593 xla_device_assignment->computation_devices(2); in TEST()
120 metadata.device_assignment().computation_devices()) { in CreateCompilationCacheKey()
114 const auto& computation_device = proto.computation_devices(computation); in Deserialize()
26 repeated ComputationDevice computation_devices = 1; field
434 repeated ComputationDevice computation_devices = 3; field