experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add dot configuration as part of module configuration.

Browse Source 
PiperOrigin-RevId: 299879284
Change-Id: I64932896316470cda7556a3618b32ac3556b3dac
This commit is contained in:
A. Unique TensorFlower 2020-03-09 10:47:13 -07:00 committed by TensorFlower Gardener
parent 8f5109a636
commit 9a7288ccf5
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/xla/service/hlo_module_config.h
View File

@ -183,6 +183,12 @@ class HloModuleConfig {
return &fusion_config_;
}
const std::vector<std::vector<int64>>& dot_config() const {
return dot_config_;
}
std::vector<std::vector<int64>>* 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<std::vector<bool>> fusion_config_;
std::vector<std::vector<int64>> dot_config_;
};
} // namespace xla