Searched refs:GetIrArrayFor (Results 1 – 2 of 2) sorted by relevance
120 llvm_ir::IrArray root_value = GetIrArrayFor(computation->root_instruction()); in EmitThreadLocalFunctionEpilogue()383 llvm_ir::EmitTupleSelect(GetIrArrayFor(tuple_select), GetIrArrayFor(pred), in HandleTupleSelect()408 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()632 llvm_ir::EmitTuple(GetIrArrayFor(sort), destination_addresses, &b_); in HandleSort()643 llvm_ir::EmitTuple(GetIrArrayFor(tuple), base_ptrs, &b_); in HandleTuple()775 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter()817 llvm_ir::IrArray source_array(GetIrArrayFor(source)); in HandleSelectAndScatter()820 llvm_ir::IrArray output_array(GetIrArrayFor(select_and_scatter)); in HandleSelectAndScatter()849 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot()850 llvm_ir::IrArray rhs_array(GetIrArrayFor(rhs)); in HandleDot()[all …]
225 llvm_ir::IrArray GetIrArrayFor(const HloInstruction* hlo);