- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for output_sharding_configuration (0.54 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
"tf_device.cluster_func"() {func = @empty_func, step_marker_location = "", num_cores_per_replica = 1 : i64} : () -> () // CHECK: input_sharding_configuration = [] // CHECK: output_sharding_configuration = [] func.return } // CHECK-LABEL: func @empty_func() { func.func @empty_func() { func.return } // ----- // Tests with a block argument inputs/outputs with no xla sharding op attached
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/tests/tpu_rewrite.mlir
// expected-error@+1 {{requires attribute 'output_sharding_configuration'}}
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/tfrt/tests/runtime_lowering_tpu.mlir
func.return } func.func @empty_func() { func.return } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
%0 = "tf_device.cluster_func"(%arg0) {_producer_name = "UNKNOWN", func = @_func, input_sharding_configuration = [""], num_cores_per_replica = 1 : i64, device_assignment = [], topology = "", output_sharding_configuration = [""], step_marker_location = "STEP_MARK_AT_ENTRY", use_spmd_for_xla_partitioning = false, use_tpu = true} : (tensor<1x3xf32>) -> (tensor<1x3xf32>) return %0 : tensor<1x3xf32> } // CHECK-LABEL: @_func
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/transforms/host_runtime/testdata/basic_cluster.mlir
func.return } func.func @empty_func() { func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 16:43:16 UTC 2023 - 639 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/testdata/malformed_cluster.mlir
func.return } func.func @empty_func() { func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 16:43:16 UTC 2023 - 742 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/runtime_lowering_gpu.mlir
func.return } func.func @empty_func() { func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 17:41:44 UTC 2023 - 840 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/testdata/spmd.mlir
= "TPU", allow_soft_placement = false, computation_shape = [], device = "", device_assignment = [0, 0, 0, 0, 0, 0, 0, 1], host_compute_core = [], input_sharding_configuration = ["{devices=[2,1]0,1}"], num_cores_per_replica = 2 : i64, output_sharding_configuration = [""], padding_map = [], step_marker_location = "STEP_MARK_AT_ENTRY", topology = "\0A\04\02\02\01\02\10\01\18\08\22 \00\00\00\00\00\00\00\01\01\00\00\00\01\00\00\01\00\01\00\00\00\01\00\01\01\01\00\00\01\01\00\01", tpu_compile_options_proto...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
namespace tensorflow { inline constexpr llvm::StringRef kInputShardingAttr = "input_sharding_configuration"; inline constexpr llvm::StringRef kOutputShardingAttr = "output_sharding_configuration"; // Parses the sharding string. This sharding string can be binary (serialized) // or human readable. mlir::LogicalResult DecodeShardingAttribute(const std::string& shard_str,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.td
%0 = "tf_device.cluster_func"(%arg0) {_xla_compile_device_type = "TPU", _replication_info = "cluster0", func = @func, num_cores_per_replica = 2, input_sharding_configuration = ["\08\01\1A\01\01\22\01\00"], output_sharding_configuration = ["\08\01\1A\01\01\22\01\00"]} : (tensor<8xi32>) -> tensor<8xi32> return %0 : tensor<8xi32> } ``` will be rewritten as: ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 10.7K bytes - Viewed (0)