Home
last modified time | relevance | path

Searched refs:conditional_mismatch_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc635 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()
Dlayout_assignment.h516 std::map<HloComputation*, ComputationLayout> conditional_mismatch_; variable