Extend the memory optimizations to also support accumulate_n ops
authorBenoit Steiner <bsteiner@google.com>
Mon, 12 Feb 2018 21:25:57 +0000 (13:25 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 12 Feb 2018 21:29:55 +0000 (13:29 -0800)
commitffc82833fff68cb16b2ffc1a0fe6077016a494e7
treea48a101b4dd1f5d1a5fd0f5a1fed9288e31e0d1d
parent8abf81de646425e910a88c385975c95f0756666b
Extend the memory optimizations to also support accumulate_n ops

PiperOrigin-RevId: 185425999
tensorflow/core/grappler/optimizers/memory_optimizer.cc