1 | /* |
2 | * SPDX-License-Identifier: Apache-2.0 |
3 | */ |
4 | |
5 | #pragma once |
6 | |
7 | #include "onnx/defs/schema.h" |
8 | |
9 | namespace ONNX_NAMESPACE { |
10 | |
11 | // Declare training operators. |
12 | |
13 | // Iterate over schema from ai.onnx.training version 1 |
14 | class OpSet_OnnxTraining_ver1 { |
15 | public: |
16 | static void ForEachSchema(std::function<void(OpSchema&&)> /* fn */) {} |
17 | }; |
18 | |
19 | // Register training operators. |
20 | inline void RegisterOnnxTrainingOperatorSetSchema() { |
21 | RegisterOpSetSchema<OpSet_OnnxTraining_ver1>(); |
22 | } |
23 | |
24 | } // namespace ONNX_NAMESPACE |
25 | |