diff --git a/tensorflow/compiler/xla/service/hlo_module_config.h b/tensorflow/compiler/xla/service/hlo_module_config.h index dee601d9e96..fa2dbc0fcad 100644 --- a/tensorflow/compiler/xla/service/hlo_module_config.h +++ b/tensorflow/compiler/xla/service/hlo_module_config.h @@ -183,6 +183,12 @@ class HloModuleConfig { return &fusion_config_; } + const std::vector>& dot_config() const { + return dot_config_; + } + + std::vector>* mutable_dot_config() { return &dot_config_; } + private: // If you add new members, be sure to update compilation_cache_key. @@ -214,6 +220,8 @@ class HloModuleConfig { FusionConfigCollection::kOff; std::vector> fusion_config_; + + std::vector> dot_config_; }; } // namespace xla