Searched refs:target_array_ (Results 1 – 1 of 1) sorted by relevance
194 const llvm_ir::IrArray& target_array_; member in xla::cpu::__anond13f39450111::DotOpEmitter216 target_array_(target_array), in DotOpEmitter()231 llvm::Value* target = target_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()313 llvm::Value* result_op = target_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()472 llvm::Type* accum_type = target_array_.GetElementLlvmType(); in EmitNaiveLlvmIrGemm()538 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()539 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()577 target_array_.EmitWriteArrayElement(/*index=*/element_index, result, b_); in EmitScalarDot()592 PrimitiveType type = target_array_.GetShape().element_type(); in EmitCallToRuntime()675 b_->CreateBitCast(target_array_.GetBasePointer(), float_ptr_type), in EmitCallToRuntime()[all …]