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