Searched refs:dispatchTable_ (Results 1 – 3 of 3) sorted by relevance
27 , dispatchTable_() in OperatorEntry()242 return dispatchTable_[dispatch_ix].isValid(); in registerKernel()395 dispatchTable_[dispatch_ix] = computeDispatchTableEntry(dispatcher, dispatch_key); in registerKernel()396 …dispatchKeyExtractor_.setOperatorHasFallthroughForKey(dispatch_key, dispatchTable_[dispatch_ix].is… in registerKernel()476 TORCH_INTERNAL_ASSERT(expected_k._equalsBoxedAndUnboxed(dispatchTable_[idx]), in registerKernel()489 if (iter == -1 || !dispatchTable_[iter].isValid()) { in registerKernel()
179 const auto& kernel = dispatchTable_[idx]; in lookup()228 std::array<KernelFunction, c10::num_runtime_entries> dispatchTable_; variable
699 # OperatorEntry::dispatchTable_ to reduce peak memory during library