Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for tile_assignment_dimensions (0.22 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir

      }
    }
    
    // -----
    
    // The following xla.OpSharding is used:
    // Proto debug string:
    //   type : OTHER
    //   tile_assignment_dimensions: 1
    //   tile_assignment_dimensions: 1
    //   tile_assignment_dimensions: 1
    //   tile_assignment_dimensions: 1
    //   tile_assignment_devices: 0
    //   last_tile_dims: REPLICATED
    // Serialized string:
    //   "\08\03\1A\06\01\01\01\01\01\01\22\01\00B\01\00"
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/tpu_validate_inputs.mlir

    //  Proto debug string:
    //    type: 	 TUPLE
    //    tuple_shardings {
    //      type: MAXIMAL
    //      tile_assignment_dimensions: 1
    //      tile_assignment_devices: 0
    //    }
    //    tuple_shardings {
    //      type: MAXIMAL
    //      tile_assignment_dimensions: 1
    //      tile_assignment_devices: 1
    //    }
    
    func.func @invalid_TUPLE_sharding_arity(%arg0: tensor<i32>) -> tensor<i32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 06:51:01 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

      }
    
      std::map<int, int> dimension_to_splits_map;
      for (int dim_index = 0; dim_index < tensor_tile_rank; ++dim_index) {
        if (sharding.tile_assignment_dimensions(dim_index) > 1) {
          dimension_to_splits_map.emplace(
              dim_index, sharding.tile_assignment_dimensions(dim_index));
        }
      }
    
      if (dimension_to_splits_map.empty()) {
        return absl::InvalidArgumentError(absl::StrCat(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

      }
    
      // Check XlaSpmdFullToShardShape's conversion from split sharding to manual
      // sharding.
      // The split sharding is:
      //   type: OTHER
      //   tile_assignment_dimensions: 2
      //   tile_assignment_dimensions: 1
      //   tile_assignment_devices: 0
      //   tile_assignment_devices: 1
      // Serialized string:
      //   "\08\03\1A\02\02\01\22\02\00\01"
      // The manual sharding is:
      //   type: MANUAL
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (1)
  5. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // Proto debug string:
      //   type: TUPLE
      //   tuple_shardings {
      //     type: MAXIMAL
      //     tile_assignment_dimensions: 1
      //     tile_assignment_devices: 0
      //   }
      //   tuple_shardings {
      //     type: MAXIMAL
      //     tile_assignment_dimensions: 1
      //     tile_assignment_devices: 0
      //   }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top