OptimizationConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/ModelCompilationConfig.h>
#include <aws/sagemaker/model/ModelQuantizationConfig.h>
#include <aws/sagemaker/model/ModelShardingConfig.h>
#include <aws/sagemaker/model/ModelSpeculativeDecodingConfig.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class OptimizationConfig {
 public:
  AWS_SAGEMAKER_API OptimizationConfig() = default;
  AWS_SAGEMAKER_API OptimizationConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API OptimizationConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const ModelQuantizationConfig& GetModelQuantizationConfig() const { return m_modelQuantizationConfig; }
  inline bool ModelQuantizationConfigHasBeenSet() const { return m_modelQuantizationConfigHasBeenSet; }
  template <typename ModelQuantizationConfigT = ModelQuantizationConfig>
  void SetModelQuantizationConfig(ModelQuantizationConfigT&& value) {
    m_modelQuantizationConfigHasBeenSet = true;
    m_modelQuantizationConfig = std::forward<ModelQuantizationConfigT>(value);
  }
  template <typename ModelQuantizationConfigT = ModelQuantizationConfig>
  OptimizationConfig& WithModelQuantizationConfig(ModelQuantizationConfigT&& value) {
    SetModelQuantizationConfig(std::forward<ModelQuantizationConfigT>(value));
    return *this;
  }
 
 
  inline const ModelCompilationConfig& GetModelCompilationConfig() const { return m_modelCompilationConfig; }
  inline bool ModelCompilationConfigHasBeenSet() const { return m_modelCompilationConfigHasBeenSet; }
  template <typename ModelCompilationConfigT = ModelCompilationConfig>
  void SetModelCompilationConfig(ModelCompilationConfigT&& value) {
    m_modelCompilationConfigHasBeenSet = true;
    m_modelCompilationConfig = std::forward<ModelCompilationConfigT>(value);
  }
  template <typename ModelCompilationConfigT = ModelCompilationConfig>
  OptimizationConfig& WithModelCompilationConfig(ModelCompilationConfigT&& value) {
    SetModelCompilationConfig(std::forward<ModelCompilationConfigT>(value));
    return *this;
  }
 
 
  inline const ModelShardingConfig& GetModelShardingConfig() const { return m_modelShardingConfig; }
  inline bool ModelShardingConfigHasBeenSet() const { return m_modelShardingConfigHasBeenSet; }
  template <typename ModelShardingConfigT = ModelShardingConfig>
  void SetModelShardingConfig(ModelShardingConfigT&& value) {
    m_modelShardingConfigHasBeenSet = true;
    m_modelShardingConfig = std::forward<ModelShardingConfigT>(value);
  }
  template <typename ModelShardingConfigT = ModelShardingConfig>
  OptimizationConfig& WithModelShardingConfig(ModelShardingConfigT&& value) {
    SetModelShardingConfig(std::forward<ModelShardingConfigT>(value));
    return *this;
  }
 
 
  inline const ModelSpeculativeDecodingConfig& GetModelSpeculativeDecodingConfig() const { return m_modelSpeculativeDecodingConfig; }
  inline bool ModelSpeculativeDecodingConfigHasBeenSet() const { return m_modelSpeculativeDecodingConfigHasBeenSet; }
  template <typename ModelSpeculativeDecodingConfigT = ModelSpeculativeDecodingConfig>
  void SetModelSpeculativeDecodingConfig(ModelSpeculativeDecodingConfigT&& value) {
    m_modelSpeculativeDecodingConfigHasBeenSet = true;
    m_modelSpeculativeDecodingConfig = std::forward<ModelSpeculativeDecodingConfigT>(value);
  }
  template <typename ModelSpeculativeDecodingConfigT = ModelSpeculativeDecodingConfig>
  OptimizationConfig& WithModelSpeculativeDecodingConfig(ModelSpeculativeDecodingConfigT&& value) {
    SetModelSpeculativeDecodingConfig(std::forward<ModelSpeculativeDecodingConfigT>(value));
    return *this;
  }
 private:
  ModelQuantizationConfig m_modelQuantizationConfig;
 
  ModelCompilationConfig m_modelCompilationConfig;
 
  ModelShardingConfig m_modelShardingConfig;
 
  ModelSpeculativeDecodingConfig m_modelSpeculativeDecodingConfig;
  bool m_modelQuantizationConfigHasBeenSet = false;
  bool m_modelCompilationConfigHasBeenSet = false;
  bool m_modelShardingConfigHasBeenSet = false;
  bool m_modelSpeculativeDecodingConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws