Commit 9f462b27 authored by Yao Zhang's avatar Yao Zhang Committed by TensorFlower Gardener
Browse files

Support reduction along axis NHW, HW, and C. One use case of "along NHW" is... 

Support reduction along axis NHW, HW, and C. One use case of "along NHW" is (unfused) batch norm. Minor code simplification for squeeze, as added transpose node has shape info now.

PiperOrigin-RevId: 180991050
parent 7cb38846
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