Home
last modified time | relevance | path

Searched refs:accum_address (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc560 llvm::Value* accum_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleDot() local
568 accum_address, // The address. in HandleDot()
585 llvm::Value* accum = Load(accum_address); in HandleDot()
603 Store(updated_accum, accum_address); in HandleDot()
627 Load(accum_address), // The value written to the target array. in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc473 llvm::Value* accum_address = in EmitNaiveLlvmIrGemm() local
481 b_->CreateStore(llvm::Constant::getNullValue(accum_type), accum_address); in EmitNaiveLlvmIrGemm()
493 llvm::Value* accum = b_->CreateLoad(accum_address); in EmitNaiveLlvmIrGemm()
512 b_->CreateStore(updated_accum, accum_address); in EmitNaiveLlvmIrGemm()
519 llvm::Value* result = b_->CreateLoad(accum_address); in EmitNaiveLlvmIrGemm()