HomePhabricator

Add `axis` attribute to the quant.stats op

Authored by Feng Liu <fengliuai@google.com> on Oct 3 2019, 8:28 PM.

Description

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

Event Timeline

A. Unique TensorFlower <gardener@tensorflow.org> committed rG8c95223e3c95: Add `axis` attribute to the quant.stats op (authored by liufengdb).Oct 3 2019, 8:29 PM