diff options
author | 2018-07-02 11:16:07 -0700 | |
---|---|---|
committer | 2018-07-02 11:18:43 -0700 | |
commit | 63d4f52e227d9dd425fdc5a885d3849d69963ff8 (patch) | |
tree | 8614ebb7657b32568abbbbaec430f288b1886795 /tensorflow/contrib/lite/toco/toco_tooling.cc | |
parent | 100995e2d0fbe56e19a6472d6f921ff0680fdfd2 (diff) |
Added graph transformation to push reshapes downstream of broadcasting binary operators if possible.
PiperOrigin-RevId: 202982286
Diffstat (limited to 'tensorflow/contrib/lite/toco/toco_tooling.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/toco_tooling.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/toco_tooling.cc b/tensorflow/contrib/lite/toco/toco_tooling.cc index a057dcef12..fc1636831b 100644 --- a/tensorflow/contrib/lite/toco/toco_tooling.cc +++ b/tensorflow/contrib/lite/toco/toco_tooling.cc @@ -79,6 +79,7 @@ void MakeGeneralGraphTransformationsSet( transformations->Add(new FuseBinaryIntoFollowingAffine); transformations->Add(new FuseBroadcastIntoFollowingBinary); transformations->Add(new MergeReshapeIntoPrecedingTranspose); + transformations->Add(new MoveBinaryOperatorBeforeReshape); transformations->Add(new ReorderElementwiseUnary); transformations->Add(new ReorderReshapeTranspose); transformations->Add(new ResolveBatchNormalization); |