Searched refs:updated_accum (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 586 llvm::Value* updated_accum; in HandleDot() local
591 updated_accum = InsertValue(accum, real_sum, {0}); in HandleDot()
594 updated_accum = InsertValue(updated_accum, imag_sum, {1}); in HandleDot()
597 updated_accum = FAdd(accum, product); in HandleDot()
601 updated_accum = Add(accum, product); in HandleDot()
603 Store(updated_accum, accum_address); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 494 llvm::Value* updated_accum; in EmitNaiveLlvmIrGemm() local
504 updated_accum = b_->CreateInsertValue( in EmitNaiveLlvmIrGemm()
506 updated_accum = b_->CreateInsertValue( in EmitNaiveLlvmIrGemm()
507 updated_accum, b_->CreateFAdd(imag(accum), product_imag), {1}); in EmitNaiveLlvmIrGemm()
510 updated_accum = b_->CreateFAdd(accum, product); in EmitNaiveLlvmIrGemm()
512 b_->CreateStore(updated_accum, accum_address); in EmitNaiveLlvmIrGemm()
|