Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SameScalesOpInterface (0.49 sec)

  1. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc

      auto spec = std::make_unique<OpQuantScaleSpec>();
      if (isa<SameScalesOpInterface>(op)) {
        spec->has_same_scale_requirement = true;
        spec->required_same_scale_func = [op](const bool sign,
                                              const int bit_width) {
          return cast<SameScalesOpInterface>(op)
              .RequiredSameOperandsAndResultsScale(sign, bit_width);
        };
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 43.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/optimize_op_order.cc

          return num_elements;
        };
    
        // If the op is the pass-through op with (3x) smaller output, the dequantize
        // op can be pushed down to the single result of this op.
        if (!llvm::dyn_cast<mlir::SameScalesOpInterface>(passthrough_op) ||
            passthrough_op->getNumResults() != 1) {
          return failure();
        }
        // Only push down the dequantize op when the output is smaller, so that it
        // can have smaller memory usage.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization.td

        InterfaceMethod<
          [{Returns the dimension index of the output channels.}],
          "int", "GetChannelDimIndex", (ins)
        >,
      ];
    }
    
    def SameOperandsAndResultsScale : OpInterface<"SameScalesOpInterface"> {
      let description = [{
        Interface for ops potentially have same operands and results scales.
      }];
    
      let methods = [
        InterfaceMethod<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 8.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

                                        rewriter))) {
                return failure();
              }
            } else if (!llvm::isa<DQ>(input.getDefiningOp()) &&
                       !llvm::isa<SameScalesOpInterface, FixedOutputRangeInterface>(
                           input.getDefiningOp())) {
              // Continue if StatisticsOp is already converted to Q-DQ pair, or
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

        if (!pre_quantized_type) return failure();
    
        Operation* def = pre_quantized.getDefiningOp();
        if (!def) return failure();
        if (llvm::isa<FixedOutputRangeInterface, SameScalesOpInterface>(def) ||
            !def->hasTrait<OpTrait::quant::QuantizableResult>()) {
          return failure();
        }
    
        // This op should not clobber def, if more than one requant of this value.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

          [F32, I64, I32, I16, I8, QI8, QUI8, UI8, UI32, I1]>:$output
      );
    
      let hasOptions = 1;
    
      let hasFolder = 1;
    
      let hasVerifier = 1;
    
      let extraClassDeclaration = [{
        // SameScalesOpInterface:
        bool RequiredSameOperandsAndResultsScale(bool sign, int bit_width) {
          // uint8 doesn't require same operands and results scales.
          bool is_uint8 = !sign && (bit_width == 8);
          return !is_uint8;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top