diff options
author | Akshay Agrawal <akshayka@google.com> | 2018-05-25 17:12:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-25 17:15:38 -0700 |
commit | 06717b77e05bd602d10fe40f4519dbb105fabd5c (patch) | |
tree | f16cf13faa46908433287b1a95204478917e486d /tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc | |
parent | e140ab8d07dd6aec70f61c0c6939506f6e67ac5e (diff) |
Provide a step container to OpKernelContexts when running eagerly.
This lets us run some ops that require step containers (e.g.
TensorArray-related ops).
Before change:
Benchmark Time(ns) CPU(ns) Iterations
-------------------------------------------------------------------
BM_CreateGraph 61292 80915 8581
BM_RunGraph 7899 13398 51251
BM_CreateAndDestroySession 2588 2594 269838
BM_KernelAndDeviceInit 2971 2976 235908
BM_KernelAndDeviceRun 505 506 1000000
After change:
Benchmark Time(ns) CPU(ns) Iterations
-------------------------------------------------------------------
BM_CreateGraph 78295 105539 8698
BM_RunGraph 9907 16988 47908
BM_CreateAndDestroySession 2773 2778 247635
BM_KernelAndDeviceInit 2678 2682 270054
BM_KernelAndDeviceRun 553 554 1000000
PiperOrigin-RevId: 198125630
Diffstat (limited to 'tensorflow/compiler/xla/service/while_loop_invariant_code_motion.cc')
0 files changed, 0 insertions, 0 deletions