- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetTileShardedOutputsToMerge (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
return mlir::failure(); } return mlir::success(); } // Collects tile sharded outputs from a tf_device.parallel_execute to remap from // the TPU computation result. mlir::LogicalResult GetTileShardedOutputsToMerge( const mlir::Location& location, const int cluster_func_output_index, llvm::ArrayRef<xla::OpSharding> output_sharding_config, llvm::SmallVector<llvm::SmallVector<int, 4>, 4> cluster_to_core_index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)