Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for 4xi16 (0.12 sec)

  1. tensorflow/compiler/mlir/lite/tests/optimize.mlir

    func.func @broadcast_to_i16_low_dim(%arg0: tensor<3xi16>, %arg1: tensor<2xi32>) -> tensor<3x3xi16> {
      %0 = "tfl.broadcast_to"(%arg0, %arg1) : (tensor<3xi16>, tensor<2xi32>) -> tensor<3x3xi16>
      return %0 : tensor<3x3xi16>
      // CHECK:  %cst = arith.constant dense<1> : tensor<3x3xi16>
      // CHECK:  %0 = tfl.mul(%arg0, %cst) <{fused_activation_function = "NONE"}> : (tensor<3xi16>, tensor<3x3xi16>) -> tensor<3x3xi16>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 284.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

      func.return %4#0, %4#1 : tensor<4x32xf32>, tensor<4x32xi32>
    }
    
    // CHECK-LABEL:   func @convert_argmin_i16(
    // CHECK-SAME:                         %[[VAL_0:.*]]: tensor<2xi16>) -> (tensor<i16>, tensor<i32>) {
    // CHECK:           %[[VAL_9:.*]] = arith.constant dense<0> : tensor<1xi32>
    // CHECK:           %[[VAL_10:.*]] = "tf.Min"(%[[VAL_0]], %[[VAL_9]]) <{keep_dims = false}> : {{.*}} -> tensor<i16>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
Back to top