Add quant.const_fake_quant_per_axis op
authorFeng Liu <fengliuai@google.com>
Mon, 9 Sep 2019 22:42:07 +0000 (15:42 -0700)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Mon, 9 Sep 2019 22:42:37 +0000 (15:42 -0700)
commitf4ae4762bf7d64d7ca46d05206955c9b44cedc49
treee1a5dc72eedc496474a62a2def3d03f089acfaf6
parentd3a6dbc0b895bdfdae3627c00d35066c4f51b032
Add quant.const_fake_quant_per_axis op

Comparing to the existing quant.const_fake_quant op, the min and max attributes
of this new op is for each channel of last dimension of the input.

PiperOrigin-RevId: 268093722
mlir/include/mlir/Dialect/QuantOps/QuantOps.td
mlir/test/Dialect/QuantOps/parse-ops.mlir