xref: /aosp_15_r20/external/pytorch/test/package/package_a/test_all_leaf_modules_tracer.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1# Owner(s): ["oncall: package/deploy"]
2
3from torch.fx import Tracer
4
5
6class TestAllLeafModulesTracer(Tracer):
7    def is_leaf_module(self, m, qualname):
8        return True
9