MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / SelectKernelMap

Method SelectKernelMap

paddle/phi/core/kernel_factory.cc:158–165  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

156}
157
158KernelKeyMap KernelFactory::SelectKernelMap(
159 const std::string& kernel_name) const {
160 auto iter = kernels_.find(kernel_name);
161 if (iter == kernels_.end()) {
162 return KernelKeyMap();
163 }
164 return iter->second;
165}
166
167bool KernelFactory::HasKernel(const std::string& kernel_name,
168 const KernelKey& kernel_key) const {

Callers 14

HasRuntimeAttributesMethod · 0.80
OpSupportGPUFunction · 0.80
SupportGPUMethod · 0.80
SupportXPUMethod · 0.80
SupportCustomDeviceMethod · 0.80
SupportsONEDNNMethod · 0.80
SupportsCUDNNMethod · 0.80
SupportsCPUBF16Method · 0.80
SupportsONEDNNFunction · 0.80
SupportsCPUBF16Function · 0.80

Calls 2

findMethod · 0.45
endMethod · 0.45

Tested by 1

TESTFunction · 0.64