Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for _collective_manager_ids (0.31 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-skip-partitioned-calls.mlir

      func.func @partitioned_call(%arg0: tensor<1x2x2x3xf32>) -> (tensor<1x2x2x3xf32>) {
        %0 = "tf.StatefulPartitionedCall"(%arg0) <{
          config = "", config_proto = "", executor_type = "", f = @some_func
        }> {
          _collective_manager_ids = [], device = ""
        } : (tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32>
        // CHECK-SKIP: tf.StatefulPartitionedCall
        // CHECK-NOSKIP: call @some_func
        // CHECK-NOSKIP-NOT: tf.StatefulPartitionedCall
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 08 20:05:12 UTC 2024
    - 1.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir

        %0 = "tf.StatefulPartitionedCall"(%arg0) <{
          config = "", config_proto = "", executor_type = "", f = @some_func
        }> {
          _collective_manager_ids = [], device = ""
        } : (tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32>
        func.return %0: tensor<1x2x2x3xf32>
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 08 20:05:12 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.mlir

          config = "", config_proto = "", executor_type = "", f = @some_func
        }> {
          _collective_manager_ids = [], device = ""
        } : (tensor<3x3xf32>, tensor<3x3xf32>) -> tensor<3x3xf32>
        %3 = "tf.PartitionedCall"(%2, %1) <{
          config = "", config_proto = "", executor_type = "", f = @some_other_func
        }> {
          _collective_manager_ids = [], device = ""
        } : (tensor<3x3xf32>, tensor<3x3xf32>) -> tensor<3x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 39.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        ```
    
        `_XlaMustCompile=true` in the following code
    
        ```mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir

    }
    
    // -----
    
    module {
    func.func @dense_image_warp(%arg0: tensor<2x4x4x1xf32>, %arg1: tensor<2x4x4x2xf32>) -> tensor<2x4x4x1xf32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 122.1K bytes
    - Viewed (0)
Back to top