Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc472 llvm::Type* accum_type = target_array_.GetElementLlvmType(); in EmitNaiveLlvmIrGemm() local
474 b_->CreateAlloca(accum_type, /*ArraySize=*/nullptr, "accum_address"); in EmitNaiveLlvmIrGemm()
481 b_->CreateStore(llvm::Constant::getNullValue(accum_type), accum_address); in EmitNaiveLlvmIrGemm()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc559 llvm::Type* accum_type = target_array.GetElementLlvmType(); in HandleDot() local
561 accum_type, // The pointee type of the alloca instruction. in HandleDot()