operator_sets_training.h#

namespace ONNX_LIGHT_NAMESPACE

namespace onnx_op

namespace training#

Typedefs

Functions

std::vector<LightOpSchema> GetAllOnnxOpTrainingSchemasWithHistory(bool init_doc = true)#

Returns the versioned schema history for all supported preview training operators (e.g. Gradient, Adam) in the ai.onnx.preview.training domain.

Preview training operators only have a single version (1); see the upstream ONNX convention which keeps preview specs at version 1 even when revised.

Returns:: Vector of LightOpSchema objects for the ai.onnx.preview.training domain, ordered by operator name and descending opset version.

Variables

constexpr const char *kOnnxPreviewTrainingDomain = "ai.onnx.preview.training"#: The ai.onnx.preview.training operator domain string.