Lines Matching refs:ir_builder_
89 converted_operands[i] = ir_builder_->CreateFPCast(
90 converted_operands[i], ir_builder_->getFloatTy());
109 result = ir_builder_->CreateFPCast(result, ir_builder_->getHalfTy());
212 return ir_builder_->CreateFDiv(llvm::ConstantFP::get(llvm_ty, 1), sqrt);
289 ir_builder_->GetInsertBlock()->getModule()->getOrInsertFunction(
296 return ir_builder_->CreateCall(callee, llvm_ir::AsArrayRef(operands));
300 llvm::Value* block_id = ir_builder_->CreateIntCast(
302 {}, {}, ir_builder_),
303 ir_builder_->getIntNTy(128), /*isSigned=*/true, "block.id");
304 llvm::Value* thread_id_in_block = ir_builder_->CreateIntCast(
306 {}, {}, ir_builder_),
307 ir_builder_->getIntNTy(128), /*isSigned=*/true, "thread.id");
308 llvm::Value* threads_per_block = ir_builder_->CreateIntCast(
310 {}, {}, ir_builder_),
311 ir_builder_->getIntNTy(128), /*isSigned=*/true, "threads_per_block");
312 return ir_builder_->CreateNSWAdd(
313 ir_builder_->CreateNSWMul(block_id, threads_per_block),
360 "reduce_window_accum_ptr", ir_builder_);
364 ir_builder_->CreateStore(init_value, accum_ptr);
367 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_);
376 SetToFirstInsertPoint(loops.GetInnerLoopBodyBasicBlock(), ir_builder_);
379 llvm::Value* in_bounds = ir_builder_->getInt1(true);
381 llvm::Value* stridden_index = ir_builder_->CreateNSWMul(
382 index[i], ir_builder_->getInt64(window.dimensions(i).stride()));
383 input_index[i] = ir_builder_->CreateNSWSub(
384 ir_builder_->CreateNSWAdd(stridden_index, window_index[i]),
385 ir_builder_->getInt64(window.dimensions(i).padding_low()));
392 in_bounds = ir_builder_->CreateAnd(
394 ir_builder_->CreateICmpULT(
396 ir_builder_->getInt64(operand->shape().dimensions(i))));
400 llvm_ir::EmitIfThenElse(in_bounds, "in_bounds", ir_builder_);
401 SetToFirstInsertPoint(if_data.true_block, ir_builder_);
409 {ir_builder_->CreateLoad(accum_ptr), input_value}));
410 ir_builder_->CreateStore(accum_value, accum_ptr);
412 SetToFirstInsertPoint(loops.GetOuterLoopExitBasicBlock(), ir_builder_);
413 return ir_builder_->CreateLoad(accum_ptr);
426 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_);
