def _ragged_size_v1(input, name=None, out_type=dtypes.int32): # pylint: disable=redefined-builtin return ragged_array_ops.size(input=input, out_type=out_type, name=name)
def testRaggedSize(self, test_input, size, ragged_rank=None): input_rt = ragged_factory_ops.constant(test_input, ragged_rank=ragged_rank) self.assertAllEqual(ragged_array_ops.size(input_rt), size)