Home
last modified time | relevance | path

Searched refs:GetElementLlvmType (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc492 result = llvm::ConstantAggregateZero::get(lhs_array.GetElementLlvmType()); in HandleDot()
559 llvm::Type* accum_type = target_array.GetElementLlvmType(); in HandleDot()
567 llvm::Constant::getNullValue(lhs_array.GetElementLlvmType()), // init 0 in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.h217 llvm::Type* GetElementLlvmType() const { return element_type_; } in GetElementLlvmType() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc472 llvm::Type* accum_type = target_array_.GetElementLlvmType(); in EmitNaiveLlvmIrGemm()
571 result = llvm::ConstantAggregateZero::get(lhs_array_.GetElementLlvmType()); in EmitScalarDot()