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