tree e9cdbfd23c25663d788eb8144f2b4607f5e91655
parent 4ede771bd96a46553a149669a5264d878f16bac1
author Ken Franko <kfranko@google.com> 1618249943 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1618250246 -0700

Refactor TPUExtractOutsideCompilation pass.

Extracts code to finer grained methods.  Passes multiple ops into some methods as preparation for handling ops with dynamic shapes that require capturing the function for shape inference.

PiperOrigin-RevId: 368037987
Change-Id: I77bdbaad5e5ab0a0a7b6c2dd781d1d4d39e072ff