Home
last modified time | relevance | path

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 Dcontrol_flow_ops_py_test.py727 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",