Searched refs:conditional_mismatch_ (Results 1 – 2 of 2) sorted by relevance
635 InsertOrDie(&conditional_mismatch_, in AddMandatoryConstraints()653 if (conditional_mismatch_.count(computation) > 0) { in AddMandatoryConstraints()655 FindOrDie(conditional_mismatch_, computation).result_layout().shape())); in AddMandatoryConstraints()1942 if (conditional_mismatch_.count(computation) > 0) { in RunOnComputation()1944 FindOrDie(conditional_mismatch_, computation).result_layout(); in RunOnComputation()2184 (i == 0 || conditional_mismatch_.count(computation) > 0) in Run()2334 conditional_mismatch_.clear(); in Init()
516 std::map<HloComputation*, ComputationLayout> conditional_mismatch_; variable