Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for PrepareLstmOutputScale (0.22 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

      // consistent. Otherwise some FileCheck tests would fail.
      RewritePatternSet patterns_1(&getContext());
      if (quant_specs_.post_training_quantization) {
        patterns_1.add<PrepareLstmOutputScale<LSTMOp>>(ctx);
        patterns_1.add<PrepareLstmOutputScale<UnidirectionalSequenceLSTMOp>>(ctx);
      }
      if (is_qdq_conversion_ ||
          quant_specs_.qdq_conversion_mode != quant::QDQConversionMode::kQDQNone) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

                           }),
            op_property->intermediates.end());
      }
      return success();
    }
    
    template <typename SourceOp>
    class PrepareLstmOutputScale : public OpRewritePattern<SourceOp> {
     public:
      explicit PrepareLstmOutputScale(MLIRContext* context)
          : OpRewritePattern<SourceOp>(context) {}
      LogicalResult matchAndRewrite(SourceOp op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
Back to top