Datum.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemakertrainingsessionruntime/SageMakerTrainingSessionRuntime_EXPORTS.h>
#include <aws/sagemakertrainingsessionruntime/model/ModelInput.h>
#include <aws/sagemakertrainingsessionruntime/model/TensorData.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMakerTrainingSessionRuntime {
namespace Model {
 
class Datum {
 public:
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API Datum() = default;
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API Datum(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API Datum& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const ModelInput& GetModelInput() const { return m_modelInput; }
  inline bool ModelInputHasBeenSet() const { return m_modelInputHasBeenSet; }
  template <typename ModelInputT = ModelInput>
  void SetModelInput(ModelInputT&& value) {
    m_modelInputHasBeenSet = true;
    m_modelInput = std::forward<ModelInputT>(value);
  }
  template <typename ModelInputT = ModelInput>
  Datum& WithModelInput(ModelInputT&& value) {
    SetModelInput(std::forward<ModelInputT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, TensorData>& GetLossFnInputs() const { return m_lossFnInputs; }
  inline bool LossFnInputsHasBeenSet() const { return m_lossFnInputsHasBeenSet; }
  template <typename LossFnInputsT = Aws::Map<Aws::String, TensorData>>
  void SetLossFnInputs(LossFnInputsT&& value) {
    m_lossFnInputsHasBeenSet = true;
    m_lossFnInputs = std::forward<LossFnInputsT>(value);
  }
  template <typename LossFnInputsT = Aws::Map<Aws::String, TensorData>>
  Datum& WithLossFnInputs(LossFnInputsT&& value) {
    SetLossFnInputs(std::forward<LossFnInputsT>(value));
    return *this;
  }
  template <typename LossFnInputsKeyT = Aws::String, typename LossFnInputsValueT = TensorData>
  Datum& AddLossFnInputs(LossFnInputsKeyT&& key, LossFnInputsValueT&& value) {
    m_lossFnInputsHasBeenSet = true;
    m_lossFnInputs.emplace(std::forward<LossFnInputsKeyT>(key), std::forward<LossFnInputsValueT>(value));
    return *this;
  }
 private:
  ModelInput m_modelInput;
 
  Aws::Map<Aws::String, TensorData> m_lossFnInputs;
  bool m_modelInputHasBeenSet = false;
  bool m_lossFnInputsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerTrainingSessionRuntime
}  // namespace Aws