Searched refs:MakeMapHlo (Results 1 – 3 of 3) sorted by relevance
145 StatusOr<HloInstruction*> MakeMapHlo(absl::Span<HloInstruction* const> operands,
302 MakeMapHlo({operand_slice_to_update, update_slice_with_dims_inserted}, in ScatterLoopBody()
301 StatusOr<HloInstruction*> MakeMapHlo(absl::Span<HloInstruction* const> operands, in MakeMapHlo() function