Searched refs:conditional_mismatch_ (Results 1 – 2 of 2) sorted by relevance
722 InsertOrDie(&conditional_mismatch_, in AddMandatoryConstraints()740 if (conditional_mismatch_.count(computation) > 0) { in AddMandatoryConstraints()742 FindOrDie(conditional_mismatch_, computation).result_layout().shape())); in AddMandatoryConstraints()2084 if (conditional_mismatch_.count(computation) > 0) { in RunOnComputation()2086 FindOrDie(conditional_mismatch_, computation).result_layout(); in RunOnComputation()2318 (i == 1 && conditional_mismatch_.count(computation) == 0) || in Run()2482 conditional_mismatch_.clear(); in Init()
537 std::map<HloComputation*, ComputationLayout> conditional_mismatch_; variable