Searched refs:output_sharding_list (Results 1 – 2 of 2) sorted by relevance
52 mlir::SmallVector<xla::OpSharding, 4>* output_sharding_list);
308 mlir::SmallVector<xla::OpSharding, 4>* output_sharding_list) { in ParseAndValidateOutputSharding() argument309 output_sharding_list->reserve(cluster_func.getNumResults()); in ParseAndValidateOutputSharding()351 output_sharding_list->emplace_back(std::move(sharding)); in ParseAndValidateOutputSharding()