Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for tile_assignment_devices_size (0.49 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top