Spaces:
Running
Running
File size: 334 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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()
|