int64_t rank = lhs_type.getRank();
    auto batch_dimensions = llvm::to_vector<4>(llvm::seq<int64_t>(0, rank - 2));
    auto lhs_contracting_dimensions = llvm::to_vector<4>(
        llvm::ArrayRef({op.getAdjX() ? rank - 2 : rank - 1}));
    auto rhs_contracting_dimensions = llvm::to_vector<4>(
        llvm::ArrayRef({op.getAdjY() ? rank - 1 : rank - 2}));

}

// CHECK-LABEL: @FuseReshapeAroundBMMNagativeTest2
// Checks that the pattern matcher FuseReshapesAroundBatchMatMulLHS does not get
// applied for this case that does not pass the constraint around input rank.
func.func @FuseReshapeAroundBMMNagativeTest2(%arg0: tensor<2x1536xf32>) -> tensor<2x768xf32> {
  %cst = arith.constant dense_resource<__elided__> : tensor<3xi32>
  %cst_0 = arith.constant dense_resource<__elided__> : tensor<2xi32>