Searched refs:meta_in_tls (Results 1 – 5 of 5) sorted by relevance
110 meta_in_tls = torch._C._meta_in_tls_dispatch_include()116 torch._C._set_meta_in_tls_dispatch_include(meta_in_tls)
666 meta_in_tls = torch._C._meta_in_tls_dispatch_include()671 torch._C._set_meta_in_tls_dispatch_include(meta_in_tls)
2241 py_module.def("_set_meta_in_tls_dispatch_include", [](bool meta_in_tls) { in initModule() argument2244 if (meta_in_tls) { in initModule()
497 meta_in_tls = torch._C._meta_in_tls_dispatch_include()498 assert meta_in_tls == prev_in_kernel, f"{meta_in_tls}, {prev_in_kernel}"
1184 def _set_meta_in_tls_dispatch_include(meta_in_tls: _bool) -> None: ...