UniformConstraints.h

Go to the documentation of this file.

//===- UniformConstraints.h - Constraints for uniform quant -----*- C++ -*-===//
//
// Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file defines a builder that lets you attach constraints necessary to
// perform a variety of uniform quantization conversions to CAG anchors.
//
//===----------------------------------------------------------------------===//

#ifndef MLIR_QUANTIZER_SUPPORT_UNIFORMCONSTRAINTS_H
#define MLIR_QUANTIZER_SUPPORT_UNIFORMCONSTRAINTS_H

#include "mlir/Quantizer/Support/Statistics.h"

namespace mlir {
namespace quantizer {

class CAGAnchorNode;
class CAGSlice;

class UniformConstraintsBuilder {
public:
  UniformConstraintsBuilder(CAGSlice &slice) : slice(slice) {}

  void coupleAnchors(CAGAnchorNode *a, CAGAnchorNode *b);

  void applyStats(CAGAnchorNode *a, TensorAxisStatistics stats);

  void clamp(CAGAnchorNode *a, APFloat minValue, APFloat maxValue);

  void propagateExplicitScale(CAGAnchorNode *from, CAGAnchorNode *to);

private:
  CAGSlice &slice;
};

} // namespace quantizer
} // namespace mlir

#endif // MLIR_QUANTIZER_SUPPORT_UNIFORMCONSTRAINTS_H