Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-02-22 10:42:45

0001 /*
0002  * SPDX-License-Identifier: Apache-2.0
0003  */
0004 
0005 #pragma once
0006 
0007 #include "onnx/defs/schema.h"
0008 
0009 namespace ONNX_NAMESPACE {
0010 
0011 // Declare training operators.
0012 
0013 // Iterate over schema from ai.onnx.training version 1
0014 class OpSet_OnnxTraining_ver1 {
0015  public:
0016   static void ForEachSchema(std::function<void(OpSchema&&)> /* fn */) {}
0017 };
0018 
0019 // Register training operators.
0020 inline void RegisterOnnxTrainingOperatorSetSchema() {
0021   RegisterOpSetSchema<OpSet_OnnxTraining_ver1>();
0022 }
0023 
0024 } // namespace ONNX_NAMESPACE