Searched refs:rhs_array_ (Results 1 – 1 of 1) sorted by relevance
225 const llvm_ir::IrArray& rhs_array_; member in xla::cpu::__anonb70c17dc0111::DotOpEmitter246 rhs_array_(rhs_array), in DotOpEmitter()257 rhs_array_.GetBasePointer()}; in EmitLinalgMatmul()347 llvm::Value* rhs = rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemm()455 swap_operands ? rhs_array_.GetBasePointer() : lhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()457 swap_operands ? lhs_array_.GetBasePointer() : rhs_array_.GetBasePointer(); in EmitTiledLlvmIrGemv()518 const Shape& rhs_shape = rhs_array_.GetShape(); in Emit()553 const Shape& rhs_shape = rhs_array_.GetShape(); in EmitNaiveLlvmIrGemm()580 rhs_array_, /*dimension_to_skip=*/rhs_reduction_dimension, "rhs"); in EmitNaiveLlvmIrGemm()634 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_); in EmitNaiveLlvmIrGemm()[all …]