Searched refs:in_kernel_invocation_manager (Results 1 – 4 of 4) sorted by relevance
25 in_kernel_invocation_manager,175 with in_kernel_invocation_manager(fake_mode):188 with in_kernel_invocation_manager(fake_mode):203 with in_kernel_invocation_manager(fake_mode):251 with in_kernel_invocation_manager(fake_mode):502 with in_kernel_invocation_manager(fake_mode):537 with in_kernel_invocation_manager(fake_mode):600 with in_kernel_invocation_manager(fake_mode):708 with in_kernel_invocation_manager(fake_mode):939 with no_python_dispatcher(), in_kernel_invocation_manager(fake_mode):
492 def in_kernel_invocation_manager( function1607 with in_kernel_invocation_manager(self), maybe_suppress():1627 with in_kernel_invocation_manager(self), maybe_suppress():1688 with in_kernel_invocation_manager(self):2012 with in_kernel_invocation_manager(self):
1531 in_kernel_invocation_manager,1537 maybe_fake_mgr = in_kernel_invocation_manager(mb_fake_mode)
31 …hem as meta tensors. This is done via a magic context manager in_kernel_invocation_manager which i…