Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for 2x3xi1 (0.09 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/tests/call_xla_module_to_stablehlo.mlir

          %outputs_9, %control_10 = tf_executor.island(%control_4) wraps "tf.Identity"(%outputs_7) {device = ""} : (tensor<2x3xi32>) -> tensor<2x3xi32>
          tf_executor.fetch %outputs_9 : tensor<2x3xi32>
        }
        return %0 : tensor<2x3xi32>
      }
    }
    
    // CHECK: module attributes
    // CHECK-SAME:  tfl.metadata = {{{.*}}keep_stablehlo_constant = "true"{{.*}}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 31 21:25:51 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/ir/QuantizeUtils.h

    /// 2. realValue is an elements attribute:
    /// (realValue: DenseElementsAttr[tensor<2x2xf32>],
    ///  quantizedElementType: UniformQuantizedType[i8:f32])
    ///   -> (DenseElementsAttr[tensor<2x2xi8>], outConvertedType: tensor<2x2xi8>)
    Attribute quantizeAttr(Attribute realValue,
                           quant::QuantizedType quantizedElementType,
                           Type &outConvertedType);
    
    /// Converts an attribute from a type based on
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jul 29 18:55:28 UTC 2022
    - 3.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass_test.cc

      module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} {
        func.func @main() -> () {
          %cst0 = "tf.Const"(){ value = dense<0> : tensor<3x5xi1>} : () -> tensor<3x5xi1>
          %0 = "tf.Where"(%cst0) : (tensor<3x5xi1>) -> tensor<?x2xi64>
          func.return
        }
      })";
    
      CellReader<int64_t> dynamism_counter(kDynamismOpCounterStreamzName);
      CellReader<int64_t> dynamism_function_counter(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 23 19:42:09 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/fold_constant_transpose.mlir

    // -----
    
    // Tests that int constants are not folded.
    
    // CHECK-LABEL: transpose_int
    func.func @transpose_int() -> tensor<3x2xi32> {
      %0 = stablehlo.constant dense<0> : tensor<2x3xi32>
      %1 = stablehlo.transpose %0, dims = [1, 0] : (tensor<2x3xi32>) -> tensor<3x2xi32>
      return %1 : tensor<3x2xi32>
    }
    // CHECK: transpose
    
    // -----
    
    // Tests that transposing an argument cannot be folded.
    
    // CHECK-LABEL: transpose_arg
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 08:06:02 UTC 2024
    - 2.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions_weight_only.mlir

    // CHECK-SAME: %[[ARG0:.+]]: tensor<1x2xf32>
    // CHECK: %[[CST:.+]] = stablehlo.constant() <{value = dense<127> : tensor<2x3xi8>}> : () -> tensor<2x3x!quant.uniform<i8<-127:127>:f32, 0.0023622048182750312>>
    // CHECK: %[[CALL:.+]] = call @quantized_dot_general_fn(%[[ARG0]], %[[CST]]) {_quantization_method = "weight_only_ptq { }"} : (tensor<1x2xf32>, tensor<2x3x!quant.uniform<i8<-127:127>:f32, 0.0023622048182750312>>) -> tensor<1x3xf32>
    // CHECK: return %[[CALL]]
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/fallback_canonicalization.mlir

      %a = corert.const_dense_tensor dense<[true, false]> : tensor<2xi1>
      %b = corert.const_dense_tensor dense<[false, true]> : tensor<2xi1>
      // CHECK: [[b:%.*]] = tfrt_fallback_async.const_dense_tensor dense<[false, true]> : tensor<2xi1>
      // CHECK-NEXT:      [[a:%.*]] = tfrt_fallback_async.const_dense_tensor dense<[true, false]> : tensor<2xi1>
      // CHECK-NEXT: tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor [[arg0]]
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Nov 04 14:07:37 UTC 2022
    - 5.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/const-values.pbtxt

            # CHECK: value = dense<{{\[\[}}1, 3, 2], [5, 4, 7]]> : tensor<2x3xi64>
          }
        }
      }
    }
    node {
      name: "z"
      op: "Const"
      device: "/device:CPU:0"
      attr {
        key: "dtype"
        value {
          type: DT_INT32
        }
      }
      attr {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 24 00:20:25 UTC 2020
    - 3.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/optimize_op_order.mlir

    func.func @no_pushdown_gather_with_no_reduction(%arg0: tensor<2xi32>) -> tensor<2x2xf32> {
      %w = "tfl.pseudo_qconst"() {qtype = tensor<2x2x!quant.uniform<i8<-127:127>:f32, 1.000000e+00>>, value = dense<127> : tensor<2x2xi8>} : () -> tensor<2x2x!quant.uniform<i8<-127:127>:f32, 1.000000e+00>>
      %dq_w = "tfl.dequantize"(%w) : (tensor<2x2x!quant.uniform<i8<-127:127>:f32, 1.000000e+00>>) -> tensor<2x2xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 01 02:06:15 UTC 2022
    - 3.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/tests/bridge/convert_tf_quant_ops_to_mhlo.mlir

      %weight_zps = "tf.Const"() { value = dense<3> : tensor<i32> } : () -> tensor<i32>
    
      // CHECK: "tf.AddV2"
      // CHECK: mhlo.constant
      // CHECK-SAME{LITERAL}: dense<[[1, 2], [3, 4]]> : tensor<2x2xi8>
      %0 = "tf.AddV2"(%input, %input) : (tensor<?x?xf32>, tensor<?x?xf32>) -> tensor<?x?xf32>
    
      // CHECK: "mhlo.dot"
      // CHECK-SAME: (tensor<?x?xf32>, tensor<2x2x!quant.uniform<i8:f32, 1.000000e+00:3>>) -> tensor<?x?xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 7.2K bytes
    - Viewed (0)
Back to top