Searched refs:get_source_fn (Results 1 – 2 of 2) sorted by relevance
751 def get_source_fn(node: torch.fx.Node): function760 self.assertEqual(get_source_fn(first_conv), get_source_fn(first_conv_bias))761 self.assertEqual(get_source_fn(second_conv), get_source_fn(second_conv_bias))764 self.assertNotEqual(get_source_fn(first_conv), get_source_fn(first_relu))765 self.assertNotEqual(get_source_fn(first_conv), get_source_fn(second_conv))766 self.assertNotEqual(get_source_fn(second_conv), get_source_fn(second_relu))767 self.assertNotEqual(get_source_fn(first_relu), get_source_fn(second_relu))770 self.assertTrue("backbone" not in get_source_fn(first_conv))771 self.assertTrue("backbone" not in get_source_fn(first_relu))772 self.assertTrue("backbone" in get_source_fn(second_conv))[all …]
134 def get_source_fn(node: torch.fx.Node) -> Optional[torch.fx.Node]: function