Searched refs:get_channel_constraints (Results 1 – 1 of 1) sorted by relevance
451 auto get_channel_constraints = [&](const HloInstruction* instruction) { in AddMandatoryConstraints() local492 CHECK(get_channel_constraints(instruction)) in AddMandatoryConstraints()495 if (!get_channel_constraints(instruction) in AddMandatoryConstraints()504 get_channel_constraints(instruction) in AddMandatoryConstraints()518 get_channel_constraints(instruction) in AddMandatoryConstraints()528 CHECK(get_channel_constraints(instruction)) in AddMandatoryConstraints()531 if (!get_channel_constraints(instruction) in AddMandatoryConstraints()539 get_channel_constraints(instruction) in AddMandatoryConstraints()1973 auto get_channel_constraints = [&](const HloInstruction* instruction) { in ConstrainChannelLayouts() local1983 get_channel_constraints(instruction) in ConstrainChannelLayouts()[all …]