Searched refs:rhs_array_ (Results 1 – 1 of 1) sorted by relevance
196 const llvm_ir::IrArray& rhs_array_; member in xla::cpu::__anond13f39450111::DotOpEmitter218 rhs_array_(rhs_array), in DotOpEmitter()230 llvm::Value* rhs = rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()315 swap_operands ? rhs_array_.GetBasePointer() : lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()317 swap_operands ? lhs_array_.GetBasePointer() : rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()378 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit()410 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()437 rhs_array_, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in EmitNaiveLlvmIrGemm()491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_); in EmitNaiveLlvmIrGemm()555 rhs_array_.EmitReadArrayElement(/*index=*/element_index, b_); in EmitScalarDot()[all …]