Update _check_shape to accept six.integer_types instead of int
Currently _check_shape requires that a shape be an `int` or sequence of `int`s. This CL allows `six.integer_type`s so now (1L,) would be a valid shape. PiperOrigin-RevId: 215589131
Loading
Please sign in to comment