: 1;
      for (int64_t i = 0; i < repeat_count; ++i) {
        tiled_inputs->push_back(split_op_output);
      }
    }
  }

  return mlir::success();
}

bool UnsupportedPartitionedShardingType(xla::OpSharding::Type sharding) {
  return sharding != xla::OpSharding::REPLICATED &&
         sharding != xla::OpSharding::OTHER;
}

}  // namespace