Searched refs:lhs_batch_idx (Results 1 – 1 of 1) sorted by relevance
163 int lhs_batch_idx, rhs_batch_idx; in createMatMulOps() local165 lhs_batch_idx = bcast.x_batch_indices()[batch_idx]; in createMatMulOps()168 lhs_batch_idx = batch_idx; in createMatMulOps()173 /*a=*/sliced_lhs[lhs_batch_idx], in createMatMulOps()