Home
last modified time | relevance | path

Searched refs:num_runtime_entries (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/c10/core/
DDispatchKeySet.cpp289 (num_runtime_entries - 1), in initializeFunctionalityOffsetsAndMasks()
291 num_runtime_entries, in initializeFunctionalityOffsetsAndMasks()
DDispatchKey.h587 constexpr uint16_t num_runtime_entries = 8; variable
589 constexpr uint16_t num_runtime_entries = num_functionality_keys + variable
DDispatchKeySet.h35 c10::num_runtime_entries < 65536,
/external/pytorch/aten/src/ATen/core/dispatch/
DOperatorEntry.h228 std::array<KernelFunction, c10::num_runtime_entries> dispatchTable_;
DOperatorEntry.cpp241 …TORCH_INTERNAL_ASSERT(dispatch_ix >= 0 && dispatch_ix < c10::num_runtime_entries, toString(k), dis… in registerKernel()
DDispatcher.h341 std::array<impl::AnnotatedKernel, num_runtime_entries> backendFallbackKernels_;
/external/pytorch/c10/test/core/
DDispatchKeySet_test.cpp382 ASSERT_EQ(count, std::ptrdiff_t{num_runtime_entries} - 1); in TEST()