Searched refs:sliced_rhs (Results 1 – 2 of 2) sorted by relevance
156 const std::vector<Value>& sliced_lhs, const std::vector<Value>& sliced_rhs, in createMatMulOps() argument174 /*b=*/sliced_rhs[rhs_batch_idx], in createMatMulOps()287 std::vector<Value> sliced_rhs = in matchAndRewrite() local293 createMatMulOps(sliced_lhs, sliced_rhs, bcast, lhs_shape[dims_a - 2], in matchAndRewrite()
47 const std::vector<Value>& sliced_rhs,