Searched refs:fusion_config_collection (Results 1 – 5 of 5) sorted by relevance
213 FusionConfigCollection fusion_config_collection() const { in fusion_config_collection() function217 FusionConfigCollection fusion_config_collection) { in set_fusion_config_collection() argument218 fusion_config_collection_ = fusion_config_collection; in set_fusion_config_collection()
107 FusionConfigCollection fusion_config_collection() const { in fusion_config_collection() function111 FusionConfigCollection fusion_config_collection) { in set_fusion_config_collection() argument112 fusion_config_collection_ = fusion_config_collection; in set_fusion_config_collection()
124 aot_options->fusion_config_collection() != FusionConfigCollection::kOff) { in CreateModuleConfig()126 aot_options->fusion_config_collection()); in CreateModuleConfig()
62 const xla::FusionConfigCollection* fusion_config_collection, in CreateModuleConfig() argument125 if (fusion_config_collection != nullptr && fusion_config != nullptr && in CreateModuleConfig()126 *fusion_config_collection != xla::FusionConfigCollection::kOff) { in CreateModuleConfig()127 config->set_fusion_config_collection(*fusion_config_collection); in CreateModuleConfig()
103 const xla::FusionConfigCollection* fusion_config_collection,