Searched refs:computation_layouts_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 555 FindOrDie(computation_layouts_, instruction->to_apply()); in AddMandatoryConstraints()
572 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints()
574 FindOrDie(computation_layouts_, condition); in AddMandatoryConstraints()
623 FindOrDie(computation_layouts_, in AddMandatoryConstraints()
630 FindOrDie(computation_layouts_, instruction->branch_computation(k)); in AddMandatoryConstraints()
634 computation_layouts_.erase(instruction->branch_computation(k)); in AddMandatoryConstraints()
895 *FindOrDie(computation_layouts_, branch_comp).mutable_parameter_layout(0) = in CopyOperandIfLayoutsDiffer()
974 FindOrDie(computation_layouts_, instruction->to_apply()))); in CheckLayouts()
985 FindOrDie(computation_layouts_, instruction->parent()))); in CheckLayouts()
993 FindOrDie(computation_layouts_, instruction->while_condition()), in CheckLayouts()
[all …]
|
| D | layout_assignment.h | 513 std::map<HloComputation*, ComputationLayout> computation_layouts_; variable
|