Searched refs:add_hlo (Results 1 – 3 of 3) sorted by relevance
182 auto add_hlo = [&](std::unique_ptr<HloInstruction> to_add) { in GetCorrectionFactor() local187 add_hlo(HloInstruction::CreateConstant(LiteralUtil::CreateR0( in GetCorrectionFactor()189 constant_factor = add_hlo(HloInstruction::CreateBroadcast( in GetCorrectionFactor()191 auto converted_partition_id = add_hlo(HloInstruction::CreateConvert( in GetCorrectionFactor()196 auto broadcast_partition_id = add_hlo(HloInstruction::CreateBroadcast( in GetCorrectionFactor()198 auto exp_operand = add_hlo( in GetCorrectionFactor()201 auto iota = add_hlo( in GetCorrectionFactor()203 exp_operand = add_hlo(HloInstruction::CreateBinary( in GetCorrectionFactor()205 return add_hlo( in GetCorrectionFactor()
2347 auto add_hlo = [&](std::unique_ptr<HloInstruction> to_add) { in HandleDynamicUpdateSlice() local2378 HloInstruction* all_dims_within_partition = add_hlo( in HandleDynamicUpdateSlice()2397 auto per_partition_size_hlo = add_hlo(HloInstruction::CreateConstant( in HandleDynamicUpdateSlice()2400 auto partition_offset = add_hlo(HloInstruction::CreateBinary( in HandleDynamicUpdateSlice()2404 auto offset_ge = add_hlo(HloInstruction::CreateCompare( in HandleDynamicUpdateSlice()2408 auto offset_lt = add_hlo(HloInstruction::CreateCompare( in HandleDynamicUpdateSlice()2410 add_hlo(HloInstruction::CreateBinary( in HandleDynamicUpdateSlice()2412 add_hlo(HloInstruction::CreateBinary( in HandleDynamicUpdateSlice()2414 add_hlo(HloInstruction::CreateConstant( in HandleDynamicUpdateSlice()2418 auto update_within_partition = add_hlo(HloInstruction::CreateBinary( in HandleDynamicUpdateSlice()[all …]
264 auto add_hlo = [&](std::unique_ptr<HloInstruction> to_add) { in PadToShape() local270 auto zero = add_hlo(HloInstruction::CreateConstant( in PadToShape()272 return add_hlo( in PadToShape()