Searched refs:toFunctionalityKey (Results 1 – 8 of 8) sorted by relevance
16 if (isPerBackendFunctionalityKey(toFunctionalityKey(k))) { in setOperatorHasFallthroughForKey()
196 auto functionality_key = toFunctionalityKey(tid); in TEST()241 auto functionality1 = toFunctionalityKey(tid1); in TEST()242 auto functionality2 = toFunctionalityKey(tid2); in TEST()
11 TORCH_CHECK(c10::toFunctionalityKey(k) == DispatchKey::Sparse, in sparseTensorSetToDeviceType()
91 return autograd_dispatch_keyset.has(toFunctionalityKey(k)); in runtimeDispatchKeySetHas()
186 << (num_backends + static_cast<uint8_t>(toFunctionalityKey(t)) - in DispatchKeySet()223 auto functionality_k = toFunctionalityKey(k); in DispatchKeySet()
215 auto fk = toFunctionalityKey(t); in toString()
661 constexpr DispatchKey toFunctionalityKey(DispatchKey k) { in toFunctionalityKey() function
677 return c10::toFunctionalityKey(k); in initDispatchBindings()