| 43 | } |
| 44 | |
| 45 | Status CoreOutOfRangeError(int core, int num_cores_per_replica) { |
| 46 | return errors::InvalidArgument( |
| 47 | "Invalid replicated core id: ", core, |
| 48 | "; num_cores_per_replica=", num_cores_per_replica); |
| 49 | } |
| 50 | } // namespace |
| 51 | |
| 52 | xla::StatusOr<absl::optional<xla::OpSharding>> ParseShardingFromDevice( |
no test coverage detected