10 namespace onnxruntime {
15 class IKernelTypeStrResolver;
34 const IKernelTypeStrResolver& kernel_type_str_resolver,
39 const IKernelTypeStrResolver& kernel_type_str_resolver) {
45 #if !defined(ORT_MINIMAL_BUILD)
48 const std::unordered_map<std::string, MLDataType>& type_constraints,
50 #endif // !defined(ORT_MINIMAL_BUILD)
52 bool IsEmpty()
const {
return kernel_creator_fn_map_.empty(); }
54 #ifdef onnxruntime_PYBIND_EXPORT_OPSCHEMA
57 return kernel_creator_fn_map_;
72 static bool VerifyKernelDef(
const Node& node,
73 const KernelDef& kernel_def,
74 const IKernelTypeStrResolver& kernel_type_str_resolver,
80 key.append(1,
' ').append(domain.empty() ?
kOnnxDomainAlias : domain).append(1,
' ').append(provider);
84 static std::string GetMapKey(
const KernelDef& kernel_def) {
85 return GetMapKey(kernel_def.OpName(), kernel_def.Domain(), kernel_def.Provider());
const std::string & ProviderType
constexpr const char * kOnnxDomainAlias
std::multimap< std::string, KernelCreateInfo > KernelCreateMap
GLsizei const GLchar *const * string
static bool HasImplementationOf(const KernelRegistry &r, const Node &node, ProviderType exec_provider, const IKernelTypeStrResolver &kernel_type_str_resolver)
basic_string_view< char > string_view
Status TryFindKernel(const Node &node, ProviderType exec_provider, const IKernelTypeStrResolver &kernel_type_str_resolver, const KernelCreateInfo **out) const
std::vector< std::pair< std::string, HashValue >> KernelDefHashes
Status Register(KernelDefBuilder &kernel_def_builder, const KernelCreateFn &kernel_creator)
std::function< Status(FuncManager &func_mgr, const OpKernelInfo &info, std::unique_ptr< OpKernel > &out)> KernelCreateFn
GT_API const UT_StringHolder version
GT_API const UT_StringHolder st