Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getRhsQuantizationAxis (0.23 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

          TF::UniformQuantizedDotHybridOpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        // Uniform Quantized type for the rhs.
        int64_t rhs_quantized_dimension = op.getRhsQuantizationAxis();
        // Currently for dot, PTQ supports per-tensor quantization.
        if (rhs_quantized_dimension != -1) {
          return rewriter.notifyMatchFailure(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

      if (failed(verify_lhs_params)) {
        return failure();
      }
    
      auto verify_rhs_params =
          VerifyScalesAndZeroPoints(op, op.getRhsScales(), op.getRhsZeroPoints(),
                                    op.getRhsQuantizationAxis());
      if (failed(verify_rhs_params)) {
        return failure();
      }
    
      return VerifyScalesAndZeroPoints(op, op.getOutputScales(),
                                       op.getOutputZeroPoints(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top