- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for tile_assignment_dimensions (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/result-sharding.mlir
// "\08\03\1A\02\01\02\22\02\00\01" // Proto debug string: // type: OTHER // tile_assignment_dimensions: 1 // tile_assignment_dimensions: 2 // tile_assignment_devices: 0 // tile_assignment_devices: 1 // // Serialized string: // "\08\01\1A\01\01\22\01\00" // Proto debug string: // type: MAXIMAL // tile_assignment_dimensions: 1 // tile_assignment_devices: 0 // // Serialized string: // ""
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 23 18:56:13 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/argument-sharding.mlir
// "\08\03\1A\02\01\02\22\02\00\01" // Proto debug string: // type: OTHER // tile_assignment_dimensions: 1 // tile_assignment_dimensions: 2 // tile_assignment_devices: 0 // tile_assignment_devices: 1 // // Serialized string: // "\08\01\1A\01\01\22\01\00" // Proto debug string: // type: MAXIMAL // tile_assignment_dimensions: 1 // tile_assignment_devices: 0 // // Serialized string: // ""
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.9K bytes - Viewed (0) -
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/transforms/host_runtime/tpu_metadata_utils_test.cc
R"pb( args { dtype: DT_FLOAT shape { unknown_rank: true } kind: PARAMETER sharding { type: OTHER tile_assignment_dimensions: 2 tile_assignment_dimensions: 1 tile_assignment_devices: 0 tile_assignment_devices: 1 } is_bounded_dynamic_dim: false } retvals { sharding {} }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 17 07:28:40 UTC 2024 - 9K 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)