const tpu::TPUCompileMetadataProto& metadata);

// Gets the proper tensor dimension from XLA OpSharding.
// "replicate_on_last_tile_dim" and "last_tile_dims" should be deducted from the
// real Tensor dimensions when tiled.
// For example:
// f32[8,512](sharding={devices=[1,1,2]0,1 last_tile_dims={REPLICATED})
// also means a replicated tensor over all devices.
//
// See xla_data.proto for detailed explanations on the fields.

}

// -----

// Tests TPIv2 with a "subgroup tiled" XLA annotation where:
//   type: OTHER
//   tile_assignment_dimensions: [4, 1, 1, 1, 2]
//   tile_assignment_devices: [0, 1, 2, 3, 4, 5, 6, 7]
//   last_tile_dims: [REPLICATED]
// Serialized string:
//   "\08\03\1A\05\04\01\01\01\02\22\08\00\01\02\03\04\05\06\07B\01\00"

// CHECK-LABEL: func @subgroup_tile_partitioned_variable

      return false;
    }
  }
  return xla_sharding.type() == xla::OpSharding::OTHER &&
         (xla_sharding.replicate_on_last_tile_dim() ||
          !xla_sharding.last_tile_dims().empty());
}

bool IsSplitSharding(const xla::OpSharding& sharding) {
  return sharding.type() == xla::OpSharding::OTHER &&
         !IsOtherReplicatedSharding(sharding);
}

//   tile_assignment_dimensions: 1
//   tile_assignment_dimensions: 1
//   tile_assignment_devices: 0
//   last_tile_dims: REPLICATED
// Serialized string:
//   "\08\03\1A\06\01\01\01\01\01\01\22\01\00B\01\00"

// Test that an input sharding with last_tile_dims REPLICATED won't generate SplitOp.
//CHECK-NOT: tf.Split