- import torch._C._lazy
- def dump(dot_file_name: str):
- """Dump TrieCache in the dot format"""
- return torch._C._lazy._dump_ir_cache(dot_file_name)
- def reset():
- """Clear TrieCache. This is needed in testing to avoid
- node reusing between different tests.
- """
- return torch._C._lazy._clear_ir_cache()
|
