カスタム演算子
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 Overflow の windows-machine-learning タグを使用してください。
- バグを報告するには、GitHub で問題を提出してください。