Commit 2956ecbb authored by Ben Barsdell's avatar Ben Barsdell
Browse files

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.
parent 9cfa96fd
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment