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