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",
|