Searched refs:GetIrArrayFor (Results 1 – 2 of 2) sorted by relevance
319 llvm_ir::EmitTupleSelect(GetIrArrayFor(tuple_select), GetIrArrayFor(pred), in HandleTupleSelect()344 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()619 llvm_ir::EmitTuple(GetIrArrayFor(sort), destination_addresses, &b_); in HandleSort()630 llvm_ir::EmitTuple(GetIrArrayFor(tuple), base_ptrs, &b_); in HandleTuple()856 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter()898 llvm_ir::IrArray source_array(GetIrArrayFor(source)); in HandleSelectAndScatter()901 llvm_ir::IrArray output_array(GetIrArrayFor(select_and_scatter)); in HandleSelectAndScatter()929 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot()930 llvm_ir::IrArray rhs_array(GetIrArrayFor(rhs)); in HandleDot()933 llvm_ir::IrArray target_array = GetIrArrayFor(dot); in HandleDot()[all …]
229 llvm_ir::IrArray GetIrArrayFor(const HloInstruction* hlo);