#pragma once

#include "LearningModelOutputs.g.h"

namespace WINML_EXPERIMENTALP {

struct LearningModelOutputs : LearningModelOutputsT<LearningModelOutputs>
{
    LearningModelOutputs(winml_experimental::LearningModelBuilder builder);

    winml_experimental::LearningModelBuilder Add(winml::ILearningModelFeatureDescriptor const& output);

    private:
    wfc::IVector<winml::ILearningModelFeatureDescriptor> output_descriptors_;
    winml_experimental::LearningModelBuilder builder_;

};

}  // namespace WINML_EXPERIMENTALP