Home
last modified time | relevance | path

Searched refs:FusionConfigCollection (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h30 enum class FusionConfigCollection { enum
213 FusionConfigCollection fusion_config_collection() const { in fusion_config_collection()
217 FusionConfigCollection fusion_config_collection) { in set_fusion_config_collection()
285 FusionConfigCollection fusion_config_collection_ =
286 FusionConfigCollection::kOff;
Dinstruction_fusion.h44 FusionConfigCollection config_collection_mode =
45 FusionConfigCollection::kOff)
138 FusionConfigCollection config_collection_mode() { in config_collection_mode()
179 FusionConfigCollection config_collection_mode_;
Dcompiler.h107 FusionConfigCollection fusion_config_collection() const { in fusion_config_collection()
111 FusionConfigCollection fusion_config_collection) { in set_fusion_config_collection()
130 FusionConfigCollection fusion_config_collection_ =
131 FusionConfigCollection::kOff;
Dinstruction_fusion.cc482 if (config_collection_mode_ != FusionConfigCollection::kOff) { in Run()
590 if (config_collection_mode_ != FusionConfigCollection::kOff) { in Run()
599 if (config_collection_mode_ != FusionConfigCollection::kOff) { in Run()
Dhlo_module_util.cc124 aot_options->fusion_config_collection() != FusionConfigCollection::kOff) { in CreateModuleConfig()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.h103 const xla::FusionConfigCollection* fusion_config_collection,
Dtpu_compile_op_support.cc62 const xla::FusionConfigCollection* fusion_config_collection, in CreateModuleConfig()
126 *fusion_config_collection != xla::FusionConfigCollection::kOff) { in CreateModuleConfig()