- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for tile_assignment_dimensions_size (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} } // namespace absl::StatusOr<std::map<int, int>> GetDimensionIndicesAndNumSplitsFromSharding( const xla::OpSharding& sharding) { int64_t tensor_tile_rank = sharding.tile_assignment_dimensions_size(); if (sharding.replicate_on_last_tile_dim()) { tensor_tile_rank--; } std::map<int, int> dimension_to_splits_map; for (int dim_index = 0; dim_index < tensor_tile_rank; ++dim_index) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
return mlir::success(); } if (RankedTensorType ranked_type = mlir::dyn_cast<RankedTensorType>(type)) { const int64_t tensor_rank = ranked_type.getRank(); int tile_assignment_rank = sharding->tile_assignment_dimensions_size(); // When a tensor is partial or subgroup tiled, its tile assignment will // have one or more dimension(s) than its rank; so, we subtract them to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)