カスタム演算子

Windows 機械学習のカスタム オペレーター Win32 API は、MLOperatorAuthor.h にあります。

API

次の内容は、カスタム演算子 API と、その構文と説明の一覧を示します。

列挙

名前 説明
MLOperatorAttributeType 属性の種類を指定します。 各属性の種類は、対応する ONNX 形式と数値的に一致します。
MLOperatorEdgeType 演算子の入力エッジまたは出力エッジの種類を指定します。
MLOperatorExecutionType カーネルが計算に CPU または GPU のいずれかを使用するか指定します。
MLOperatorKernelOptions カスタム演算子カーネルを登録するときに使用するオプションを指定します。
MLOperatorParameterOptions 演算子の入力エッジと出力エッジのオプション フラグを指定します。
MLOperatorSchemaEdgeTypeFormat 入力エッジと出力エッジの種類を記述する方法を指定します。
MLOperatorTensorDataType テンソルのデータ型を指定します。 各データ型は、対応する ONNX 形式と数値的に一致します。

関数

名前 説明
MLCreateOperatorRegistry カスタム演算子カーネルとカスタム演算子スキーマの登録に使用できる IMLOperatorRegistry のインスタンスを作成します。

インターフェイス

名前 説明
IMLOperatorAttributes 演算子を使用したモデルによって決定されるとおり、演算子の属性の値を表します。
IMLOperatorKernel カスタム演算子カーネルによって実装されます。
IMLOperatorKernelContext カーネルの計算中にオペレーターの使用状況に関する情報を提供します。
IMLOperatorKernelCreationContext カーネルの作成中にオペレーターの使用状況に関する情報を提供します。
IMLOperatorKernelFactory カスタム演算子カーネルの作成者が、そのカーネルのインスタンスを作成するために実装されました。
IMLOperatorRegistry カスタム演算子カーネルとスキーマのレジストリのインスタンスを表します。
IMLOperatorShapeInferenceContext 図形推論子が呼び出されている間にオペレーターの使用状況に関する情報を提供します。
IMLOperatorShapeInferrer 演算子の出力テンソル エッジの図形を推論するため、図形推論子によって実装されます。
IMLOperatorTensor カスタム演算子カーネルの計算中に使用されるテンソルの表現。
IMLOperatorTensorShapeDescription 演算子の入力と出力テンソル図形のセットを表します。
IMLOperatorTypeInferenceContext 型推論子が呼び出されている間にオペレーターの使用状況に関する情報を提供します。
IMLOperatorTypeInferrer 演算子の出力エッジの型を推論するため、型推論子によって実装されます。

構造体

名前 説明
MLOperatorAttribute カスタム演算子の属性の名前とプロパティを指定します。
MLOperatorAttributeNameValue カスタム演算子の属性の名前と値を指定します。
MLOperatorEdgeDescription 演算子の入力エッジまたは出力エッジのプロパティを指定します。
MLOperatorEdgeTypeConstraint カスタム演算子カーネルとスキーマでサポートされるエッジの種類に対する制約を指定します。
MLOperatorKernelDescription そのスキーマの登録に使用されるカスタム演算子カーネルの説明。
MLOperatorSchemaDescription そのスキーマの登録に使用されるカスタム演算子スキーマの説明。
MLOperatorSchemaEdgeDescription 演算子の入力エッジまたは出力エッジに関する情報を指定します。
MLOperatorSetId 演算子セットの ID を指定します。

Note

Windows ML に関するヘルプについては、次のリソースを参照してください。

  • Windows ML に関する技術的な質問をしたり、質問に回答したりするには、Stack Overflowwindows-machine-learning タグを使用してください。
  • バグを報告するには、GitHub で問題を提出してください。