Skip to content
Commit 8c95223e authored by Feng Liu's avatar Feng Liu Committed by A. Unique TensorFlower
Browse files

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
parent 0dfa7fc9
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment