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

Remove HostModule.

Browse Source 
PiperOrigin-RevId: 209640734
This commit is contained in:
Tong Shen 2018-08-21 12:32:07 -07:00 committed by TensorFlower Gardener
parent 237cb7f7d0
commit c61a49ec31
2 changed files with 1 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -72,15 +72,6 @@ class HloModuleConfig {
return debug_options_.xla_hlo_profile();
}
// Sets/returns whether this is a "host module". Host modules are used to
// record the data- and control-flow dependencies of host side computation
// that communicates with compiled code. They are used for analysis and
// scheduling purposes, but no code is generated.
bool is_host_module() const { return is_host_module_; }
void set_is_host_module(bool is_host_module) {
is_host_module_ = is_host_module;
}
// Sets/returns the module seed set during execution.
void set_seed(uint64 seed) { seed_ = seed; }
uint64 seed() const { return seed_; }

5
tensorflow/compiler/xla/service/hlo_module_group_metadata.cc
View File

@ -287,10 +287,7 @@ tensorflow::gtl::optional<int64> HloModuleGroupMetadata::GetInstructionDevice(
}
int64 HloModuleGroupMetadata::GetDeviceModulesCount() const {
return std::count_if(modules_.begin(), modules_.end(),
[](const HloModule* module) {
return !module->config().is_host_module();
});
return modules_.size();
}
Status HloModuleGroupMetadata::RecordInstructions() {