- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for MoveSharding (0.08 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
SetReplicatedOrMaximalShardingIfNoShardingFound(logical_device_vec, use_spmd, optional_output_sharding); TF_RETURN_IF_ERROR(MoveSharding(optional_input_sharding, input_sharding)); TF_RETURN_IF_ERROR(MoveSharding(optional_output_sharding, output_sharding)); return absl::OkStatus(); } // Extracts input/output sharding configuration of `cluster_func` by parsing
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)