Add `axis` attribute to the quant.stats op
authorFeng Liu <fengliuai@google.com>
Fri, 4 Oct 2019 03:28:40 +0000 (20:28 -0700)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 4 Oct 2019 03:29:08 +0000 (20:29 -0700)
commit8c95223e3c9555165fb9f56db35c3c8e85ddd4c1
treeb27dd86cd68a61eba456735a5e3c90103008e902
parent0dfa7fc9082cd7792f3769f738468d5f1dfd5a3a
Add `axis` attribute to the quant.stats op

The first dim length of the axisStats attribute should equals to the slice size
of the input argument when splitted by the axis dimension.

PiperOrigin-RevId: 272798042
mlir/include/mlir/Dialect/QuantOps/QuantOps.td
mlir/lib/Dialect/QuantOps/IR/QuantOps.cpp
mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp
mlir/test/Dialect/QuantOps/parse-ops-invalid.mlir
mlir/test/Dialect/QuantOps/parse-ops.mlir