Method AssignDevice

tensorflow/compiler/xla/service/hlo_sharding.cc:29–31

using absl::StrJoin;

HloSharding HloSharding::AssignDevice(int64 device_id) {
  return HloSharding(device_id);
}

HloSharding HloSharding::Tile1D(const Shape& input_shape, int64 num_tiles) {
  CHECK_EQ(1, input_shape.rank());