Searched refs:log_torchscript_usage (Results 1 – 4 of 4) sorted by relevance
16 from torch._utils_internal import log_torchscript_usage80 log_torchscript_usage("save", model_id=_get_model_id(m))171 log_torchscript_usage("load", model_id=_get_model_id(ret))
22 from torch._utils_internal import log_torchscript_usage1438 log_torchscript_usage("script", model_id=_get_model_id(ret))
999 log_torchscript_usage,1016 log_torchscript_usage("trace", model_id=_get_model_id(traced_func))
139 def log_torchscript_usage(api: str, **kwargs): function