Searched refs:module_exists (Results 1 – 1 of 1) sorted by relevance
3153 def module_exists(gm: GraphModule, path: str) -> bool: function3165 self.assertTrue(module_exists(a, "net_b.net_c.dropout"))3171 self.assertTrue(module_exists(a, "net_b.net_c.conv"))3183 self.assertFalse(module_exists(a, "net_b.net_c.conv"))3244 self.assertFalse(module_exists(a, "net_b.embedding"))3245 self.assertFalse(module_exists(a, "net_b.net_c.embedding"))3246 self.assertFalse(module_exists(a, "net_b.net_c.rnn"))3247 self.assertFalse(module_exists(a, "batch_norm_2d"))