Searched refs:autograd_dispatch_keyset_with_ADInplaceOrView (Results 1 – 3 of 3) sorted by relevance
106 dispatch_key_guard_(c10::autograd_dispatch_keyset_with_ADInplaceOrView) { in AutoDispatchBelowADInplaceOrView()
684 constexpr DispatchKeySet autograd_dispatch_keyset_with_ADInplaceOrView = variable914 return (s - autograd_dispatch_keyset_with_ADInplaceOrView - in legacyExtractDispatchKey()
171 key_set_ = key_set - c10::autograd_dispatch_keyset_with_ADInplaceOrView; in TensorImpl()