Upravit

Sdílet prostřednictvím


Custom operators

The Windows Machine Learning custom operator Win32 APIs are located in MLOperatorAuthor.h.

APIs

The following is a list of the custom operator APIs with their syntax and descriptions.

Enumerations

Name Description
MLOperatorAttributeType Specifies the type of an attribute. Each attribute type numerically matches the corresponding ONNX type.
MLOperatorEdgeType Specifies the types of an input or output edge of an operator.
MLOperatorExecutionType Specifies whether a kernel uses the CPU or GPU for computation.
MLOperatorKernelOptions Specifies options used when registering custom operator kernels.
MLOperatorParameterOptions Specifies option flags of input and output edges of operators.
MLOperatorSchemaEdgeTypeFormat Specifies the manner in which types of input and output edges are described.
MLOperatorTensorDataType Specifies the data type of a tensor. Each data type numerically matches the corresponding ONNX type.

Functions

Name Description
MLCreateOperatorRegistry Creates an instance of IMLOperatorRegistry which may be used to register a custom operator kernel and custom operator schema.

Interfaces

Name Description
IMLOperatorAttributes Represents the values of an operator's attributes, as determined by a model using the operator.
IMLOperatorKernel Implemented by custom operator kernels.
IMLOperatorKernelContext Provides information about an operator's usage while kernels are being computed.
IMLOperatorKernelCreationContext Provides information about an operator's usage while kernels are being created.
IMLOperatorKernelFactory Implemented by the author of a custom operator kernel to create instances of that kernel.
IMLOperatorRegistry Represents an instance of a registry for the custom operator kernel and schema.
IMLOperatorShapeInferenceContext Provides information about an operator's usage while shape inferrers are being invoked.
IMLOperatorShapeInferrer Implemented by shape inferrers to infer shapes of an operator's output tensor edges.
IMLOperatorTensor Representation of a tensor used during computation of custom operator kernels.
IMLOperatorTensorShapeDescription Represents the set of input and output tensor shapes of an operator.
IMLOperatorTypeInferenceContext Provides information about an operator's usage while type inferrers are being invoked.
IMLOperatorTypeInferrer Implemented by type inferrers to infer the types of an operator's output edges.

Structures

Name Description
MLOperatorAttribute Specifies the name and properties of an attribute of a custom operator.
MLOperatorAttributeNameValue Specifies the name and value(s) of an attribute of a custom operator.
MLOperatorEdgeDescription Specifies the properties of an input or output edge of an operator.
MLOperatorEdgeTypeConstraint Specifies constraints upon the types of edges supported in custom operator kernels and schema.
MLOperatorKernelDescription Description of a custom operator kernel used to register that schema.
MLOperatorSchemaDescription Description of a custom operator schema used to register that schema.
MLOperatorSchemaEdgeDescription Specifies information about an input or output edge of an operator.
MLOperatorSetId Specifies the identity of an operator set.

Note

Use the following resources for help with Windows ML:

  • To ask or answer technical questions about Windows ML, please use the windows-machine-learning tag on Stack Overflow.
  • To report a bug, please file an issue on our GitHub.