Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 14 of 14 for 1x2x3x4x5xf32 (0.15 sec)

  1. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      %0 = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<*xi32>) -> tensor<7x8x1x2x3x4x5x6xf32>
      func.return %0: tensor<7x8x1x2x3x4x5x6xf32>
    
    // CHECK-LABEL: broadcast_to_high_dim_with_unknown_shape
    // CHECK:  [[BCT:%.*]] = "tf.BroadcastTo"(%arg0, %arg1) : (tensor<1x2x3x4x5x6xf32>, tensor<*xi32>) -> tensor<7x8x1x2x3x4x5x6xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK:           %[[VAL_8:.*]] = "tf.Select"(%[[VAL_2]], %[[VAL_3]], %[[VAL_7]]) : (tensor<1x2x3x4xi1>, tensor<1x2x3x4xf32>, tensor<1x2x3x4xf32>) -> tensor<1x2x3x4xf32>
    // CHECK:           return %[[VAL_8]] : tensor<1x2x3x4xf32>
    // CHECK:         }
    func.func @sign(%arg0: tensor<1x2x3x4xf32>, %arg1: tensor<1x2x3x4xf32>) -> tensor<1x2x3x4xf32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir

    }
    
    // CHECK-LABEL: func @BiasAddGrad_NHWC
    func.func @BiasAddGrad_NHWC(%arg0: tensor<2x3x4x5xf32>) -> tensor<5xf32> {
      // CHECK: "tf.Const"() <{value = dense<[0, 1, 2]> : tensor<3xi64>}>
      // CHECK: "tf.Sum"({{.*}}) <{keep_dims = false}>
    
      %0 = "tf.BiasAddGrad"(%arg0) {data_format = "NHWC"} : (tensor<2x3x4x5xf32>) -> tensor<5xf32>
      func.return %0 : tensor<5xf32>
    }
    
    // CHECK-LABEL: func @BiasAddGrad_NCHW
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 05 18:35:42 UTC 2024
    - 92K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    // -----
    
    // CHECK-LABEL: func @sign
    // CHECK-SAME: [[ARG:%arg.*]]: tensor<1x2x3x4xf32>
    func.func @sign(%arg0: tensor<1x2x3x4xf32>) -> tensor<1x2x3x4xf32> {
      // CHECK: [[SIGN:%.*]] = mhlo.sign [[ARG]]
      // CHECK: return [[SIGN]] : tensor<1x2x3x4xf32>
      %0 = "tf.Sign"(%arg0) : (tensor<1x2x3x4xf32>) -> (tensor<1x2x3x4xf32>)
      func.return %0 : tensor<1x2x3x4xf32>
    }
    
    // -----
    
    // CHECK-LABEL: func @sign_dynamic
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top