def METHOD_NAME():
    verify_batch_matmul(1, 1, 16, 16, 32)
    verify_batch_matmul(5, 5, 16, 16, 32)
    verify_batch_matmul(5, 5, 16, 20, 32)
    verify_batch_matmul(30, 30, 16, 20, 32)
    # Test batch broadcasting.
    verify_batch_matmul(1, 5, 16, 16, 32)
    verify_batch_matmul(5, 1, 16, 16, 32)
    # Test dynamic batch
    verify_batch_matmul(1, 1, 16, 16, 32, dynamic=True)
    verify_batch_matmul(5, 5, 16, 16, 32, dynamic=True)