Searched refs:include_infra_modes (Results 1 – 3 of 3) sorted by relevance
31 def is_in_torch_dispatch_mode(include_infra_modes=True) -> bool: argument32 … return _is_in_torch_dispatch_mode if include_infra_modes else _is_in_non_infra_torch_dispatch_mode
1765 self.assertFalse(is_in_torch_dispatch_mode(include_infra_modes=False))1768 self.assertTrue(is_in_torch_dispatch_mode(include_infra_modes=False))1772 is_in_torch_dispatch_mode(include_infra_modes=False)1776 self.assertTrue(is_in_torch_dispatch_mode(include_infra_modes=False))1778 self.assertFalse(is_in_torch_dispatch_mode(include_infra_modes=False))1781 self.assertFalse(is_in_torch_dispatch_mode(include_infra_modes=False))
1218 is_in_torch_dispatch_mode(include_infra_modes=False)1229 elif is_in_torch_dispatch_mode(include_infra_modes=False):