Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for getQuantizationAxis (0.35 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

        // must match for lhs, rhs, and output.
        activation_quantization_axis =
            GetQuantizationAxis(rewriter, op, /*operand_index=*/0);
        if (activation_quantization_axis == rewriter.getI64IntegerAttr(-1)) {
          activation_quantization_axis =
              GetQuantizationAxis(rewriter, op, /*operand_index=*/1);
        }
      }
      attrs.push_back(rewriter.getNamedAttr("lhs_quantization_axis",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

            op, op.getOutput().getType(), op.getScales(), op.getZeroPoints(),
            /*expressed_type=*/rewriter.getF32Type(), op.getQuantizationMinVal(),
            op.getQuantizationMaxVal(), op.getQuantizationAxis(), rewriter);
        if (failed(output_type)) {
          return failure();
        }
    
        auto result = rewriter.create<mhlo::UniformQuantizeOp>(
            op->getLoc(), *output_type, op.getInput());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //
    
    LogicalResult UniformQuantizeOp::verify() {
      UniformQuantizeOp op = *this;
      return VerifyScalesAndZeroPoints(op, op.getScales(), op.getZeroPoints(),
                                       op.getQuantizationAxis());
    }
    
    //===----------------------------------------------------------------------===//
    // UniformRequantizeOp
    //===----------------------------------------------------------------------===//
    //
    
    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