SampleOperationConfig.h

 
#pragma once
#include <aws/sagemakertrainingsessionruntime/SageMakerTrainingSessionRuntime_EXPORTS.h>
#include <aws/sagemakertrainingsessionruntime/model/ComputeLogprobsConfig.h>
#include <aws/sagemakertrainingsessionruntime/model/LoadWeightsForSamplingConfig.h>
#include <aws/sagemakertrainingsessionruntime/model/SampleConfig.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMakerTrainingSessionRuntime {
namespace Model {
 
class SampleOperationConfig {
 public:
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API SampleOperationConfig() = default;
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API SampleOperationConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API SampleOperationConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERTRAININGSESSIONRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const SampleConfig& GetSample() const { return m_sample; }
  inline bool SampleHasBeenSet() const { return m_sampleHasBeenSet; }
  template <typename SampleT = SampleConfig>
  void SetSample(SampleT&& value) {
    m_sampleHasBeenSet = true;
    m_sample = std::forward<SampleT>(value);
  }
  template <typename SampleT = SampleConfig>
  SampleOperationConfig& WithSample(SampleT&& value) {
    SetSample(std::forward<SampleT>(value));
    return *this;
  }
 
 
  inline const ComputeLogprobsConfig& GetComputeLogprobs() const { return m_computeLogprobs; }
  inline bool ComputeLogprobsHasBeenSet() const { return m_computeLogprobsHasBeenSet; }
  template <typename ComputeLogprobsT = ComputeLogprobsConfig>
  void SetComputeLogprobs(ComputeLogprobsT&& value) {
    m_computeLogprobsHasBeenSet = true;
    m_computeLogprobs = std::forward<ComputeLogprobsT>(value);
  }
  template <typename ComputeLogprobsT = ComputeLogprobsConfig>
  SampleOperationConfig& WithComputeLogprobs(ComputeLogprobsT&& value) {
    SetComputeLogprobs(std::forward<ComputeLogprobsT>(value));
    return *this;
  }
 
 
  inline const LoadWeightsForSamplingConfig& GetLoadWeightsForSampling() const { return m_loadWeightsForSampling; }
  inline bool LoadWeightsForSamplingHasBeenSet() const { return m_loadWeightsForSamplingHasBeenSet; }
  template <typename LoadWeightsForSamplingT = LoadWeightsForSamplingConfig>
  void SetLoadWeightsForSampling(LoadWeightsForSamplingT&& value) {
    m_loadWeightsForSamplingHasBeenSet = true;
    m_loadWeightsForSampling = std::forward<LoadWeightsForSamplingT>(value);
  }
  template <typename LoadWeightsForSamplingT = LoadWeightsForSamplingConfig>
  SampleOperationConfig& WithLoadWeightsForSampling(LoadWeightsForSamplingT&& value) {
    SetLoadWeightsForSampling(std::forward<LoadWeightsForSamplingT>(value));
    return *this;
  }
 private:
  SampleConfig m_sample;
 
  ComputeLogprobsConfig m_computeLogprobs;
 
  LoadWeightsForSamplingConfig m_loadWeightsForSampling;
  bool m_sampleHasBeenSet = false;
  bool m_computeLogprobsHasBeenSet = false;
  bool m_loadWeightsForSamplingHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerTrainingSessionRuntime
}  // namespace Aws