Learn more  » Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Bower components Debian packages RPM packages NuGet packages

edgify / torch   python

Repository URL to install this package:

Version: 2.0.1+cpu 

/ _lazy / config.py

import torch._C._lazy


def get_force_fallback():
    """Get the config used to force LTC fallback"""
    return torch._C._lazy._get_force_fallback()


def set_force_fallback(configval):
    """Set the config used to force LTC fallback"""
    torch._C._lazy._set_force_fallback(configval)


def set_reuse_ir(val: bool):
    """Set the config to reuse IR nodes for faster tracing"""
    torch._C._lazy._set_reuse_ir(val)