SetReplicatedOrMaximalShardingIfNoShardingFound(logical_device_vec, use_spmd,
                                                  optional_output_sharding);
  TF_RETURN_IF_ERROR(MoveSharding(optional_input_sharding, input_sharding));
  TF_RETURN_IF_ERROR(MoveSharding(optional_output_sharding, output_sharding));

  return absl::OkStatus();
}

// Extracts input/output sharding configuration of `cluster_func` by parsing