// type but not replicated.
bool IsSplitSharding(const xla::OpSharding& sharding);

// Returns whether the sharding is replicated. It includes sharding with
// REPLICATED type and replicated OTHER type.
bool IsReplicatedSharding(const xla::OpSharding& sharding);

// Returns a map of dimension indices and number of splits for tiled sharding.
absl::StatusOr<std::map<int, int>> GetDimensionIndicesAndNumSplitsFromSharding(