- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for input_sharding_configuration (0.48 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
func.func @check_sharding_attrs_exists_for_empty_cluster_func() { "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 } // -----
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
func.return } func.func @empty_func() { func.return } } // ----- // Tests `tf_device.cluster_func` with missing `input_sharding_configuration` attribute.
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/ifrt/rewrite_cluster_to_ifrt_call.mlir
%outputs = "tf.TPUCompilationResult"() {_tpu_compilation_status = "cluster", device = ""} : () -> tensor<!tf_type.string>
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/utils/xla_sharding_util.h
#include "xla/xla_data.pb.h" #include "tensorflow/core/protobuf/tpu/compile_metadata.pb.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.
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
func @tf_tpu_rewrite(%arg0: tensor<8xi32>) -> tensor<8xi32> { %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:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
attributes, and `func` receives the mhlo.sharding attribute: ```mlir func @test(%arg0: tensor<*xi32>) { %0 = "tf_device.cluster_func"(%arg0) { func = @func, input_sharding_configuration = ["\01\02\03"], output_sharding_configuration = ["\08\01\1A\01\01\22\01\00"], step_marker_location = ""} : (tensor<*xi32>) -> tensor<*xi32> return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0)