tree a4ea9d2a2b31d8efdf2f6136999c515637021b7a
parent 56aa1b17ed7f5315850574d81082a91d04547efc
author Xiao Yu <fishx@google.com> 1596225394 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1596227060 -0700

Introduce Eager placer for TFRT for op handler placement. This placer reuse placement logic borrowed from TF eager runtime.

We need to build NodeDef in order to reuse placement logic from existing runtime. As a result, we update SetAttrs.* methods to set fallback attribute for both fallback ops and native ops. Verify with a few existing benchmark (Resnet50 GPU, Kumamon CPU, tf.add micro benchmark on cpu) that it does not cause noticeable overheads if user specify the device explicitly. If the user does not specify device, it will cause ~1us overhead to build node_def.

PiperOrigin-RevId: 324267375
Change-Id: Iaf7e29495e469de37666bb2df646d1ee44aba13c