- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetDimensionIndicesAndNumSplitsFromSharding (0.72 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
// Split nodes at ith depth from the original input node represent nodes // that split the input data at i-th dimension. auto dimension_to_splits_map = GetDimensionIndicesAndNumSplitsFromSharding(input_sharding); if (!dimension_to_splits_map.ok()) { LOG(ERROR) << dimension_to_splits_map.status(); return mlir::failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)