Searched refs:get_channel_constraints (Results 1 – 1 of 1) sorted by relevance
435 auto get_channel_constraints = [&](const HloInstruction* instruction) { in AddMandatoryConstraints() local478 CHECK(get_channel_constraints(instruction)) in AddMandatoryConstraints()481 if (!get_channel_constraints(instruction) in AddMandatoryConstraints()490 get_channel_constraints(instruction) in AddMandatoryConstraints()503 Shape new_shape = get_channel_constraints(instruction) in AddMandatoryConstraints()510 CHECK(get_channel_constraints(instruction)) in AddMandatoryConstraints()513 if (!get_channel_constraints(instruction) in AddMandatoryConstraints()521 get_channel_constraints(instruction) in AddMandatoryConstraints()1811 auto get_channel_constraints = [&](const HloInstruction* instruction) { in ConstrainChannelLayouts() local1821 get_channel_constraints(instruction) in ConstrainChannelLayouts()[all …]