Searched refs:target_array_ (Results 1 – 1 of 1) sorted by relevance
223 const llvm_ir::IrArray& target_array_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter244 target_array_(target_array), in DotOpEmitter()258 llvm::Value* target_ptr = target_array_.GetBasePointer(); in EmitLinalgMatmul()348 llvm::Value* target = target_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()453 llvm::Value* result_op = target_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()615 llvm::Type* accum_type = target_array_.GetElementLlvmType(); in EmitNaiveLlvmIrGemm()687 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()688 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()726 target_array_.EmitWriteArrayElement(/*index=*/element_index, result, b_); in EmitScalarDot()741 PrimitiveType type = target_array_.GetShape().element_type(); in EmitCallToRuntime()[all …]