Home
last modified time | relevance | path

Searched refs:switch_to_thread_local (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dpadding_fifo_queue_test.py120 ops.get_default_graph().switch_to_thread_local()
149 ops.get_default_graph().switch_to_thread_local()
190 ops.get_default_graph().switch_to_thread_local()
638 ops.get_default_graph().switch_to_thread_local()
660 ops.get_default_graph().switch_to_thread_local()
685 ops.get_default_graph().switch_to_thread_local()
712 ops.get_default_graph().switch_to_thread_local()
818 ops.get_default_graph().switch_to_thread_local()
848 ops.get_default_graph().switch_to_thread_local()
921 ops.get_default_graph().switch_to_thread_local()
[all …]
H A Dfifo_queue_test.py773 ops.get_default_graph().switch_to_thread_local()
802 ops.get_default_graph().switch_to_thread_local()
829 ops.get_default_graph().switch_to_thread_local()
983 ops.get_default_graph().switch_to_thread_local()
1012 ops.get_default_graph().switch_to_thread_local()
1035 ops.get_default_graph().switch_to_thread_local()
1063 ops.get_default_graph().switch_to_thread_local()
1183 ops.get_default_graph().switch_to_thread_local()
1206 ops.get_default_graph().switch_to_thread_local()
1256 ops.get_default_graph().switch_to_thread_local()
[all …]
H A Dpriority_queue_test.py71 ops.get_default_graph().switch_to_thread_local()
120 ops.get_default_graph().switch_to_thread_local()
173 ops.get_default_graph().switch_to_thread_local()
232 ops.get_default_graph().switch_to_thread_local()
H A Dconditional_accumulator_test.py403 ops.get_default_graph().switch_to_thread_local()
432 ops.get_default_graph().switch_to_thread_local()
468 ops.get_default_graph().switch_to_thread_local()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/
H A Ddense_update_ops_no_tsan_test.py35 ops.get_default_graph().switch_to_thread_local()
65 ops.get_default_graph().switch_to_thread_local()
100 ops.get_default_graph().switch_to_thread_local()
130 ops.get_default_graph().switch_to_thread_local()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_conditional_accumulator_test.py271 ops.get_default_graph().switch_to_thread_local()
306 ops.get_default_graph().switch_to_thread_local()
344 ops.get_default_graph().switch_to_thread_local()
387 ops.get_default_graph().switch_to_thread_local()
426 ops.get_default_graph().switch_to_thread_local()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-graph.pbtxt134 name: "switch_to_thread_local"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.-graph.pbtxt134 name: "switch_to_thread_local"
H A Dtensorflow.__internal__.-func-graph.pbtxt235 name: "switch_to_thread_local"
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dmirrored_run.py171 ops.get_default_graph().switch_to_thread_local()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/random/
H A Drandom_shuffle_queue_test.py43 ops.get_default_graph().switch_to_thread_local()
1241 ops.get_default_graph().switch_to_thread_local()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dops_test.py1955 g.switch_to_thread_local()
2003 g.switch_to_thread_local()
2067 g.switch_to_thread_local()
H A Dops.py3219 self.switch_to_thread_local()
5334 def switch_to_thread_local(self): member in Graph