Searched refs:BaseTorchFunctionMode (Results 1 – 4 of 4) sorted by relevance
12 from torch.overrides import _get_current_function_mode_stack, BaseTorchFunctionMode65 class TestMode1(BaseTorchFunctionMode):68 class TestMode2(BaseTorchFunctionMode):98 class IgnoredMode(BaseTorchFunctionMode):124 with BaseTorchFunctionMode(), BaseTorchFunctionMode(), BaseTorchFunctionMode():132 with BaseTorchFunctionMode(), IgnoredMode(), BaseTorchFunctionMode():138 with IgnoredMode(), IgnoredMode(), BaseTorchFunctionMode(), BaseTorchFunctionMode():169 m = BaseTorchFunctionMode()170 m1 = BaseTorchFunctionMode()197 m = BaseTorchFunctionMode()[all …]
87 from torch.overrides import BaseTorchFunctionMode89 f_ctxs.append(BaseTorchFunctionMode)
25 BaseTorchFunctionMode1630 base_mode = BaseTorchFunctionMode()
2079 class BaseTorchFunctionMode(TorchFunctionMode): class