Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for restrict_scale (0.29 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

        const std::vector<int>& tensors = lstm_property.restrict_scale[0];
        if (tensors.size() != 2) {
          op.emitError(
              "Unexpected restricted_scale from operator property."
              " Should only have a pair of indices.");
          return failure();
        }
        return processRestrictScale(op, tensors[0], tensors[1], rewriter);
      }
    
     private:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

      // it is disabled explicitly.
      bool infer_tensor_range =
          (quant_specs_.post_training_quantization || eager_quantize) &&
          !quant_specs_.disable_infer_tensor_range;
    
      // LSTM's restrict_scale requirement should be handled before converting stats
      // to Q-DQ ops. The pattern is applied for non-PTQ case to make op ordering
      // consistent. Otherwise some FileCheck tests would fail.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
Back to top