- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for tile_assignment_dimensions (0.38 sec)
-
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/tests/tpu_sharding_identification.mlir
// The following xla.OpSharding is used: // Proto debug string: // type : OTHER // tile_assignment_dimensions: 1 // tile_assignment_dimensions: 1 // tile_assignment_dimensions: 2 // tile_assignment_devices: 0 // tile_assignment_devices: 1 // last_tile_dims: REPLICATED // Serialized string: // "\08\03\1A\03\01\01\02\22\02\00\01B\01\00"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 20 19:07:52 UTC 2024 - 47.5K 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/tensorflow/tests/tpu_resource_partitioning.mlir
// code that uses a TPUPartitionedInputV2 without REPLICATED sharding. // The TPUParitionedInput has the following OpSharding: // Proto debug string: // input 0 // type: OTHER // tile_assignment_dimensions: 1 // tile_assignment_dimensions: 2 // tile_assignment_devices: 0 // tile_assignment_devices: 1 // Serialized string: // "\08\03\1A\02\01\02\22\02\00\01"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 15.7K 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-communication.mlir
// if there are more than one operand or more than one result. // // The following op sharding assigns full tensors to TPU core 0. // Proto debug string: // type: MAXIMAL // tile_assignment_dimensions: 1 // tile_assignment_devices: 0 // Serialized string: // "\08\01\1A\01\01\22\01\00" // CHECK-LABEL: func @host_compute // CHECK-SAME: ([[ARG0:%.*]]: tensor<i32>, [[ARG1:%.*]]: tensor<i64>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 08 18:24:20 UTC 2024 - 38.3K bytes - Viewed (0)