include_training_cases.h

namespace ONNX_LIGHT_NAMESPACE

namespace onnx_backend_test

Functions

void RegisterAdamCases(std::vector<TestCase> &registry)

Registers the ai.onnx.preview.training::Adam backend test node case(s).

void CollectTrainingTestCases(std::vector<TestCase> &registry)

Collects all training op category backend test node cases by invoking every Register*Cases helper declared in this header.