projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 21ab9e8) | patch
Initial Python API for specifying outside_compilation blocks that call out from a...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Sat, 7 Apr 2018 00:31:43 +0000 (17:31 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sat, 7 Apr 2018 01:16:27 +0000 (18:16 -0700)
commit38d1ac1e4f5b2a6e88eee43d332292898e0afc41
tree70185d4cbf776694f64963ca4a45df4a3ae4d090tree | snapshot
parent21ab9e8aa4b541b40ab27b394d390b08ea54a6d8commit | diff
Initial Python API for specifying outside_compilation blocks that call out from a TPU computation.

For now outside_compilation cannot occur inside any compiled control flow (while loop or conditional). If the computation is replicated, the outside_compilation ops are also replicated. Both of these restrictions will be lifted in followup CLs.

PiperOrigin-RevId: 191963758
tensorflow/compiler/tf2xla/functionalize_control_flow.cc diff | blob | history
tensorflow/contrib/tpu/python/tpu/tpu.py diff | blob | history
tensorflow/contrib/tpu/python/tpu/tpu_test.py diff | blob | history
tensorflow/python/eager/function.py diff | blob | history
tensorflow/python/framework/ops.py diff | blob | history
tensorflow/python/ops/control_flow_ops.py diff | blob | history
tensorflow/python/ops/gradients_impl.py diff | blob | history
Domain: Machine Learning / ML Framework;