Make eager functions runable on TPU
authorIgor Ganichev <iga@google.com>
Tue, 8 May 2018 00:21:53 +0000 (17:21 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 8 May 2018 01:28:37 +0000 (18:28 -0700)
commitb6906d19bacffa25fec074216a5c281e5689ef03
tree10abe83c46aecb367fb789d6bf70e48d2b01f88a
parent482ed8eb666d8bc1e5c3f47e5c1e61cc19e0fdb1
Make eager functions runable on TPU

PiperOrigin-RevId: 195748721
13 files changed:
tensorflow/compiler/jit/BUILD
tensorflow/compiler/jit/create_xla_launch_op.cc
tensorflow/compiler/jit/create_xla_launch_op.h [new file with mode: 0644]
tensorflow/compiler/jit/create_xla_launch_op_test.cc [new file with mode: 0644]
tensorflow/compiler/jit/kernels/xla_launch_op.cc
tensorflow/compiler/jit/kernels/xla_launch_op.h
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
tensorflow/compiler/jit/xla_launch_util.cc
tensorflow/compiler/jit/xla_launch_util.h
tensorflow/compiler/tests/BUILD
tensorflow/compiler/tests/eager_test.py
tensorflow/contrib/eager/python/examples/resnet50/resnet50_test.py
tensorflow/python/eager/function.py