- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for tile_assignment_dimensions (0.22 sec)
-
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) -
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) -
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) -
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) -
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)