Searched refs:changes_to_bf16_ (Results 1 – 2 of 2) sorted by relevance
82 auto it = changes_to_bf16_.find(inst); in RevertIfFusionInternalBF16Changes()83 return it != changes_to_bf16_.end() && !it->second.empty(); in RevertIfFusionInternalBF16Changes()89 auto root_changes_it = changes_to_bf16_.find(root); in RevertIfFusionInternalBF16Changes()90 if (root_changes_it != changes_to_bf16_.end()) { in RevertIfFusionInternalBF16Changes()141 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()160 changes_to_bf16_.erase(inst); in RevertIfFusionInternalBF16Changes()841 changes_to_bf16_.clear(); in Run()915 for (auto& change : changes_to_bf16_) { in Run()989 auto it = changes_to_bf16_.find(hlo); in OutputTypeAfterChange()990 if (it == changes_to_bf16_.end()) { in OutputTypeAfterChange()[all …]
206 changes_to_bf16_; variable