Searched refs:changes_to_bf16_ (Results 1 – 2 of 2) sorted by relevance
80 auto it = changes_to_bf16_.find(inst); in RevertIfFusionInternalBF16Changes()81 return it != changes_to_bf16_.end() && !it->second.empty(); in RevertIfFusionInternalBF16Changes()87 auto root_changes_it = changes_to_bf16_.find(root); in RevertIfFusionInternalBF16Changes()88 if (root_changes_it != changes_to_bf16_.end()) { in RevertIfFusionInternalBF16Changes()139 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()158 changes_to_bf16_.erase(inst); in RevertIfFusionInternalBF16Changes()721 changes_to_bf16_.clear(); in Run()795 for (auto& change : changes_to_bf16_) { in Run()835 auto it = changes_to_bf16_.find(hlo); in OutputTypeAfterChange()836 if (it == changes_to_bf16_.end()) { in OutputTypeAfterChange()[all …]
201 changes_to_bf16_; variable