Searched refs:_count_matching_switch_nodes_on_device (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
H A D | control_flow_ops_py_test.py | 727 def _count_matching_switch_nodes_on_device(self, run_metadata, device_str, member in ControlFlowTest 773 self._count_matching_switch_nodes_on_device(run_metadata, "CPU", 776 self._count_matching_switch_nodes_on_device(run_metadata, "GPU", 811 self._count_matching_switch_nodes_on_device(run_metadata, "CPU", 814 self._count_matching_switch_nodes_on_device(run_metadata, "GPU", 847 self._count_matching_switch_nodes_on_device(run_metadata, "CPU", 850 self._count_matching_switch_nodes_on_device(run_metadata, "GPU",
|