Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for input_sharding_configuration (0.22 sec)

  1. 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)
  2. 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)
Back to top