Modify PadInsertion pass so that it matches other passes.
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 18 May 2018 14:47:41 +0000 (07:47 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 18 May 2018 14:50:02 +0000 (07:50 -0700)
commit56b466583339e4bb110572a0b48b46b42d11e8eb
treebbde4307dd19900d246cb31aeb6e03b412f559db
parent2934484b3a4802c3b4644e6fc9a2b1c647d2eb9a
Modify PadInsertion pass so that it matches other passes.

Currently, PadInsertion only iterates over the instructions in the
entry_computation. Other passes iterate over MakeNonfusionComputations.
When we run on HloSnapshots derived from TPU benchmarks, this makes a
difference, because it seems none of the convolutions are inside the entry
computation.

PiperOrigin-RevId: 197145067
tensorflow/compiler/xla/service/gpu/pad_insertion.cc
tensorflow/compiler/xla/service/gpu/pad_insertion.h