Searched refs:_count_matching_switch_nodes_on_device (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 679 def _count_matching_switch_nodes_on_device(self, run_metadata, device_str): member in ControlFlowTest
718 self._count_matching_switch_nodes_on_device(run_metadata, "CPU"), 1)
720 self._count_matching_switch_nodes_on_device(run_metadata, "GPU"), 0)
746 self._count_matching_switch_nodes_on_device(run_metadata, "CPU"), 0)
748 self._count_matching_switch_nodes_on_device(run_metadata, "GPU"), 1)
|