OptimizationConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/ModelQuantizationConfig.h>
#include <aws/sagemaker/model/ModelCompilationConfig.h>
#include <aws/sagemaker/model/ModelShardingConfig.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;}
  private:
 
    ModelQuantizationConfig m_modelQuantizationConfig;
    bool m_modelQuantizationConfigHasBeenSet = false;
 
    ModelCompilationConfig m_modelCompilationConfig;
    bool m_modelCompilationConfigHasBeenSet = false;
 
    ModelShardingConfig m_modelShardingConfig;
    bool m_modelShardingConfigHasBeenSet = false;
  };
 
} // namespace Model
} // namespace SageMaker
} // namespace Aws