Searched refs:rewrite_with_while (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | functional_ops_test.py | 836 def _tfSum(self, use_gpu, rewrite_with_while): argument
847 1, 21, 1, [0.], Body, rewrite_with_while=rewrite_with_while)[0],
850 100, 0, -1, [0.], Body, rewrite_with_while=rewrite_with_while)
874 1, 21, 1, [0.], TestBody, rewrite_with_while=True)[0]
899 for rewrite_with_while in (True, False):
900 use_gpu = not rewrite_with_while
904 rewrite_with_while=rewrite_with_while)
907 rewrite_with_while=rewrite_with_while)
910 rewrite_with_while=rewrite_with_while)
916 "While" if rewrite_with_while else "For"))
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | functional_ops.py | 1076 rewrite_with_while=None): argument
1097 if rewrite_with_while:
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/examples/convert/ |
| H A D | operation_fusion.md | 250 rewrite_with_while=compiled)
|