Home
last modified time | relevance | path

Searched refs:SearchTFProfNode (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/profiler/
Dprofiler_test.py128 self.assertNotEqual(lib.SearchTFProfNode(pb1, 'DW'), None)
129 self.assertEqual(lib.SearchTFProfNode(pb1, 'DW2'), None)
130 self.assertEqual(lib.SearchTFProfNode(pb1, 'add'), None)
140 self.assertNotEqual(lib.SearchTFProfNode(pb2, 'DW'), None)
141 self.assertNotEqual(lib.SearchTFProfNode(pb2, 'DW2'), None)
142 self.assertEqual(lib.SearchTFProfNode(pb2, 'add'), None)
152 self.assertNotEqual(lib.SearchTFProfNode(pb3, 'DW'), None)
153 self.assertNotEqual(lib.SearchTFProfNode(pb3, 'DW2'), None)
154 self.assertNotEqual(lib.SearchTFProfNode(pb3, 'add'), None)
156 self.assertEqual(lib.SearchTFProfNode(pb0, 'Conv2D'), None)
[all …]
Dmodel_analyzer_test.py739 n = lib.SearchTFProfNode(ret, 'mul')
740 n2 = lib.SearchTFProfNode(ret2, 'mul')
/external/tensorflow/tensorflow/python/profiler/internal/
Dmodel_analyzer_testlib.py95 def SearchTFProfNode(node, name): function
100 r = SearchTFProfNode(c, name)
Drun_metadata_test.py195 mm_node = lib.SearchTFProfNode(