Searched refs:computation_layouts_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 537 FindOrDie(computation_layouts_, instruction->to_apply()); in AddMandatoryConstraints()
554 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints()
556 FindOrDie(computation_layouts_, condition); in AddMandatoryConstraints()
605 FindOrDie(computation_layouts_, in AddMandatoryConstraints()
612 FindOrDie(computation_layouts_, instruction->branch_computation(j)); in AddMandatoryConstraints()
918 FindOrDie(computation_layouts_, instruction->to_apply()))); in CheckLayouts()
929 FindOrDie(computation_layouts_, instruction->parent()))); in CheckLayouts()
937 FindOrDie(computation_layouts_, instruction->while_condition()), in CheckLayouts()
938 FindOrDie(computation_layouts_, instruction->while_body()))); in CheckLayouts()
944 FindOrDie(computation_layouts_, branch_computation)); in CheckLayouts()
[all …]
|
| D | layout_assignment.h | 505 std::map<HloComputation*, ComputationLayout> computation_layouts_; variable
|