Add layout optimization for reduce ops w/ keepdims
- Allows NHWC -> NCHW transposes to be propagated through reduce ops that have the attribute keepdims=true. This avoids redundant transposes at the end of some models such as Slim's resnet50.
Loading
Please sign in to comment