Home
last modified time | relevance | path

Searched refs:nonFallthroughKeysPerBackend_ (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/core/dispatch/
DDispatchKeyExtractor.cpp23 …SSERT(backend_idx >= 0 && static_cast<uint8_t>(backend_idx) < nonFallthroughKeysPerBackend_.size()… in setOperatorHasFallthroughForKey()
25nonFallthroughKeysPerBackend_[backend_idx] = nonFallthroughKeysPerBackend_[backend_idx].remove(k); in setOperatorHasFallthroughForKey()
27nonFallthroughKeysPerBackend_[backend_idx] = nonFallthroughKeysPerBackend_[backend_idx].add(k); in setOperatorHasFallthroughForKey()
31 for (const auto i : c10::irange(nonFallthroughKeysPerBackend_.size() - 1)) { in setOperatorHasFallthroughForKey()
32 if (nonFallthroughKeysPerBackend_[i] != nonFallthroughKeysPerBackend_[i+1]) { in setOperatorHasFallthroughForKey()
46 for (const auto i : c10::irange(nonFallthroughKeysPerBackend_.size())) { in setOperatorHasFallthroughForKey()
47 nonFallthroughKeysPerBackend_[i] = nonFallthroughKeysPerBackend_[i].remove(k); in setOperatorHasFallthroughForKey()
50 for (const auto i : c10::irange(nonFallthroughKeysPerBackend_.size())) { in setOperatorHasFallthroughForKey()
51 nonFallthroughKeysPerBackend_[i] = nonFallthroughKeysPerBackend_[i].add(k); in setOperatorHasFallthroughForKey()
DDispatchKeyExtractor.h170 return impl::computeDispatchKeySet(ks, nonFallthroughKeysPerBackend_[backend_idx]); in getDispatchKeySetBoxed()
182 return impl::computeDispatchKeySet(ks, nonFallthroughKeysPerBackend_[backend_idx]); in getDispatchKeySetUnboxed()
217 for (const auto i : c10::irange(nonFallthroughKeysPerBackend_.size())) { in DispatchKeyExtractor()
218 nonFallthroughKeysPerBackend_[i] = DispatchKeySet::FULL; in DispatchKeyExtractor()
236 std::array<DispatchKeySet, num_backends> nonFallthroughKeysPerBackend_; member