Searched refs:rhs_multi_index (Results 1 – 3 of 3) sorted by relevance
534 std::vector<llvm::Value*> rhs_multi_index = in HandleDot() local542 rhs_multi_index[i] = lhs_multi_index[i]; in HandleDot()554 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()582 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_array.GetShape(), in HandleDot()
435 std::vector<llvm::Value*> rhs_multi_index = in EmitNaiveLlvmIrGemm() local460 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()461 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_shape, in EmitNaiveLlvmIrGemm()
2152 std::vector<llvm::Value*> lhs_multi_index, rhs_multi_index; in EmitElementalDot() local2163 rhs_multi_index.push_back( in EmitElementalDot()2167 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()2169 rhs_multi_index.insert(rhs_multi_index.begin() + rhs_contracting_dim, in EmitElementalDot()2171 IrArray::Index rhs_index(rhs_multi_index, hlo->operand(1)->shape(), in EmitElementalDot()