Searched refs:rhs_batch_idx (Results 1 – 1 of 1) sorted by relevance
163 int lhs_batch_idx, rhs_batch_idx; in createMatMulOps() local166 rhs_batch_idx = bcast.y_batch_indices()[batch_idx]; in createMatMulOps()169 rhs_batch_idx = batch_idx; in createMatMulOps()174 /*b=*/sliced_rhs[rhs_batch_idx], in createMatMulOps()