Searched refs:backendFallbackKernels_ (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | Dispatcher.cpp | 69 , backendFallbackKernels_() in Dispatcher()
425 TORCH_CHECK(idx >= 0 && static_cast<uint64_t>(idx) < backendFallbackKernels_.size(), "idx=", idx); in registerFallback()
427 !backendFallbackKernels_[idx].kernel.isValid(), in registerFallback()
429 backendFallbackKernels_[idx].debug, ", new registration ", debug in registerFallback()
433 …backendFallbackKernels_[idx] = impl::AnnotatedKernel(std::move(kernel), nullptr, std::move(debug)); in registerFallback()
450 backendFallbackKernels_[idx] = {}; in deregisterFallback_()
|
| H A D | Dispatcher.h | 194 return backendFallbackKernels_[dispatch_ix].kernel.isValid(); in hasBackendFallbackForDispatchKey()
341 std::array<impl::AnnotatedKernel, num_runtime_entries> backendFallbackKernels_; variable
763 return backendFallbackKernels_[idx].kernel;
|
| H A D | OperatorEntry.cpp | 376 if (dispatcher.backendFallbackKernels_[dispatch_ix].kernel.isValid()) { in registerKernel()
377 return {dispatcher.backendFallbackKernels_[dispatch_ix], "backend fallback"}; in registerKernel()
|