Searched refs:rhs_multi_index (Results 1 – 2 of 2) sorted by relevance
578 std::vector<llvm::Value*> rhs_multi_index = in EmitNaiveLlvmIrGemm() local603 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()604 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_shape, in EmitNaiveLlvmIrGemm()
2211 std::vector<llvm::Value*> lhs_multi_index, rhs_multi_index; in EmitElementalDot() local2222 rhs_multi_index.push_back( in EmitElementalDot()2226 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()2228 rhs_multi_index.insert(rhs_multi_index.begin() + rhs_contracting_dim, in EmitElementalDot()2230 IrArray::Index rhs_index(rhs_multi_index, hlo->operand(1)->shape(), in EmitElementalDot()