Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 54 for uniform_quantize (0.32 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/shape_inference_with_shape_specialization.mlir

        // CHECK-NEXT: %[[UQ:.*]] = "tf.UniformQuantize"(%arg0, %cst, %cst_0) <{quantization_axis = -1 : i64, quantization_max_val = 127 : i64, quantization_min_val = -128 : i64}> : (tensor<1xf32>, tensor<f32>, tensor<i32>) -> tensor<1x!tf_type.qint8>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 2.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

      %zp = "tf.Const"() { value = dense<3> : tensor<i32> } : () -> tensor<i32>
      %0 = "tf.UniformQuantize"(%arg0, %scale, %zp) {
        quantization_axis = -1 : i64,
        quantization_min_val = -128 : i64,
        quantization_max_val = 127 : i64
      } : (tensor<10xf32>, tensor<f32>, tensor<i32>) -> tensor<10x!tf_type.qint8>
      %1 = "tf.UniformDequantize"(%0, %scale, %zp) {
        quantization_axis = -1 : i64,
        quantization_min_val = -128 : i64,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 01:03:21 UTC 2024
    - 35.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc

          %0 = "tf.UniformQuantize"(%arg0, %scales, %zps) {
            quantization_axis = -1 : i64, quantization_min_val = -2147483648 : i64, quantization_max_val = 2147483647 : i64
          } : (tensor<2x2xf32>, tensor<f32>, tensor<i32>) -> tensor<2x2x!tf_type.qint32>
          %qmax = "tf.UniformQuantize"(%max, %scales, %zps) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 18:43:55 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

    }
    
    // Returns kernel input feature dimension of TFLite convolutions.
    int64_t GetConvolutionKernelInputFeatureDimension(bool is_depthwise) {
      return is_depthwise ? 0 : 3;
    }
    
    // stablehlo.uniform_quantize -> tfl.quantize
    // TODO: b/322428814 - Add StableHLO quantizer integration tests for ODML.
    class RewriteUniformQuantizeOp
        : public OpRewritePattern<stablehlo::UniformQuantizeOp> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_uniform_quantized.mlir

      // Quantize initial input at the start of the graph. Output is qint8.
      func.func @quantize_i8(%input : tensor<*xf32>, %input_scale : tensor<*xf32>, %input_zp : tensor<*xi32>) -> tensor<*x!tf_type.qint8> {
        %quantize = "tf.UniformQuantize"(%input, %input_scale, %input_zp) {
          Tin = "tfdtype$DT_FLOAT",
          Tout = "tfdtype$DT_QINT8",
          quantization_axis = -1,
          quantization_min_val = -128,
          quantization_max_val = 127,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 29 01:13:58 UTC 2023
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc

              clEnumValN(OpSet::TF, "TF",
                         "Uses TF ops that mimic quantization behavior"),
              clEnumValN(OpSet::XLA, "XLA", "Uses TF XLA ops"),
              clEnumValN(OpSet::UNIFORM_QUANTIZED, "UNIFORM_QUANTIZED",
                         "Uses TF Uniform Quantized ops"))};
    };
    
    llvm::StringRef InsertQuantizedFunctionsPass::GetFunctionLibrary(
        QuantMethod quantization_method, OpSet op_set) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

              'target_opset': quant_opts_pb2.UNIFORM_QUANTIZED,
              'input_shape_dynamic': False,
              'enable_per_channel_quantization': False,
          },
          {
              'testcase_name': 'dilation_with_bias_and_bn_and_relu6_to_uq',
              'activation_fn': nn_ops.relu6,
              'has_bias': True,
              'has_batch_norm': True,
              'target_opset': quant_opts_pb2.UNIFORM_QUANTIZED,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc

          *this, "target-opset", llvm::cl::init(OpSet::UNIFORM_QUANTIZED),
          llvm::cl::desc("Choose target opset."),
          llvm::cl::values(
              clEnumValN(OpSet::TF, "TF",
                         "Uses TF ops that mimic quantization behavior"),
              clEnumValN(OpSet::XLA, "XLA", "Uses TF XLA ops"),
              clEnumValN(OpSet::UNIFORM_QUANTIZED, "UNIFORM_QUANTIZED",
                         "Uses TF Uniform Quantized ops"))};
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc

              clEnumValN(OpSet::TF, "TF",
                         "Uses TF ops that mimic quantization behavior"),
              clEnumValN(OpSet::XLA, "XLA", "Uses TF XLA ops"),
              clEnumValN(OpSet::UNIFORM_QUANTIZED, "UNIFORM_QUANTIZED",
                         "Uses TF Uniform Quantized ops"))};
    
      Option<int64_t> min_num_elements_for_weights_{
          *this, "min-num-elements-for-weights", llvm::cl::init(0),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/quantization/ir/QuantizeUtils.cc

        Attribute realValue, quant::QuantizedType quantizedElementType,
        Type &outConvertedType) {
      if (auto uniformQuantized =
              mlir::dyn_cast<quant::UniformQuantizedType>(quantizedElementType)) {
        UniformQuantizedValueConverter converter(uniformQuantized);
        return quantizeAttrUniform(realValue, uniformQuantized, converter,
                                   outConvertedType);
      }
      if (auto uniformQuantizedPerAxis =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.1K bytes
    - Viewed (0)
Back to top