1# mypy: allow-untyped-defs 2import torch._C._lazy 3 4 5def dump(dot_file_name: str): 6 """Dump TrieCache in the dot format""" 7 return torch._C._lazy._dump_ir_cache(dot_file_name) 8 9 10def reset(): 11 """Clear TrieCache. This is needed in testing to avoid 12 node reusing between different tests. 13 """ 14 return torch._C._lazy._clear_ir_cache() 15