- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for tile_assignment_devices_size (0.49 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/xla_sharding_util_test.cc
EXPECT_EQ(0, sharding.tile_assignment_devices_size()); } TEST(DecodeShardingAttributeTest, CheckMaximalShardString) { xla::OpSharding sharding; EXPECT_TRUE( DecodeShardingAttribute("{maximal device=0}", sharding).succeeded()); EXPECT_TRUE(sharding.type() == sharding.MAXIMAL); EXPECT_EQ(1, sharding.tile_assignment_devices_size()); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
if (sharding.type() == xla::OpSharding::OTHER && sharding.tile_assignment_devices_size() != num_cores_per_replica) return cluster_func.emitError(llvm::formatv( "incorrect sharding format for outputs. Number of " "tiled outputs({0}) must match the number of logical " "devices({1})", sharding.tile_assignment_devices_size(), num_cores_per_replica));
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/tensorflow/transforms/tpu_validate_inputs.cc
if (sharding.type() != xla::OpSharding::MAXIMAL) { continue; } if (sharding.tile_assignment_devices_size() != 1) { op->emitOpError("TF/XLA TPU bridge input check: There must be ") << "exactly 1 device for MAXIMAL sharding." << " Number of devices assigned are " << sharding.tile_assignment_devices_size() << "\n"; return false; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0)