Searched refs:lhs_batch (Results 1 – 2 of 2) sorted by relevance
519 (lhs_contract, rhs_contract), (lhs_batch, rhs_batch) = dimension_numbers523 dot_dims_proto.lhs_batch_dimensions.extend(lhs_batch)
835 const int64 lhs_batch = first_operand->shape().dimensions(new_batch_dim); in CanPropagate() local845 if (rhs_batch * kNumSplits != lhs_batch) { in CanPropagate()858 const int64 lhs_batch = activations->shape().dimensions( in CanPropagate() local873 if (rhs_batch != kNumSplits * lhs_batch) { in CanPropagate()