diff --git a/tensorflow/python/ops/ragged/BUILD b/tensorflow/python/ops/ragged/BUILD index d88543c400f2432ea620ccddcab983337abe3fc2..c0db8bfbb5cc5bd20a7bad3e69866df506b7dc00 100644 --- a/tensorflow/python/ops/ragged/BUILD +++ b/tensorflow/python/ops/ragged/BUILD @@ -62,6 +62,7 @@ py_library( "//tensorflow/python:ragged_array_ops_gen", "//tensorflow/python:tensor_shape", "//tensorflow/python:tensor_util", + "//tensorflow/python:util", ], ) @@ -82,6 +83,7 @@ py_library( "//tensorflow/python:ragged_conversion_ops_gen", "//tensorflow/python:sparse_tensor", "//tensorflow/python:tensor_shape", + "//tensorflow/python:util", ], ) @@ -95,6 +97,7 @@ py_library( "//tensorflow/python:framework_ops", "//tensorflow/python:math_ops", "//tensorflow/python:tensor_util", + "//tensorflow/python:util", "//tensorflow/python/ops/ragged:ragged_tensor", "//tensorflow/python/ops/ragged:ragged_tensor_value", "//third_party/py/numpy", @@ -110,6 +113,7 @@ py_library( ":ragged_tensor", ":ragged_util", "//tensorflow/python:framework_ops", + "//tensorflow/python:util", ], ) @@ -147,6 +151,7 @@ py_library( "//tensorflow/python:math_ops", "//tensorflow/python:ragged_math_ops_gen", "//tensorflow/python:tensor_util", + "//tensorflow/python:util", ], ) @@ -189,6 +194,7 @@ py_library( "//tensorflow/python:framework_ops", "//tensorflow/python:session", "//tensorflow/python:tensor_shape", + "//tensorflow/python:util", ], ) @@ -216,7 +222,10 @@ py_library( name = "ragged_tensor_value", srcs = ["ragged_tensor_value.py"], srcs_version = "PY2AND3", - deps = ["//third_party/py/numpy"], + deps = [ + "//tensorflow/python:util", + "//third_party/py/numpy", + ], ) py_library( @@ -245,6 +254,7 @@ py_library( "//tensorflow/python:math_ops", "//tensorflow/python:tensor_shape", "//tensorflow/python:tensor_util", + "//tensorflow/python:util", ], ) diff --git a/tensorflow/python/ops/ragged/ragged_factory_ops.py b/tensorflow/python/ops/ragged/ragged_factory_ops.py index 2c63e1c7994c31b6ed53e37e65498a843e2bb595..695accc6529278d59532c351b2f6c1e57986c9b5 100644 --- a/tensorflow/python/ops/ragged/ragged_factory_ops.py +++ b/tensorflow/python/ops/ragged/ragged_factory_ops.py @@ -24,11 +24,13 @@ from tensorflow.python.framework import constant_op from tensorflow.python.framework import ops from tensorflow.python.ops.ragged import ragged_tensor from tensorflow.python.ops.ragged import ragged_tensor_value +from tensorflow.python.util.tf_export import tf_export #=============================================================================== # Op to construct a constant RaggedTensor from a nested Python list. #=============================================================================== +@tf_export("ragged.constant") def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, name=None): """Constructs a constant RaggedTensor from a nested Python list. @@ -74,6 +76,7 @@ def constant(pylist, dtype=None, ragged_rank=None, inner_shape=None, name=None): inner_shape) +@tf_export(v1=["ragged.constant_value"]) def constant_value(pylist, dtype=None, ragged_rank=None, inner_shape=None): """Constructs a RaggedTensorValue from a nested Python list. diff --git a/tensorflow/python/ops/ragged/ragged_functional_ops.py b/tensorflow/python/ops/ragged/ragged_functional_ops.py index 751f2c73592c676d0dd5eec4f9dc45430cd646b1..7344c96465ca3ea1c3befa67af70ad4113de3ad8 100644 --- a/tensorflow/python/ops/ragged/ragged_functional_ops.py +++ b/tensorflow/python/ops/ragged/ragged_functional_ops.py @@ -21,8 +21,10 @@ from __future__ import print_function from tensorflow.python.framework import ops from tensorflow.python.ops.ragged import ragged_tensor from tensorflow.python.ops.ragged import ragged_util +from tensorflow.python.util.tf_export import tf_export +@tf_export("ragged.map_flat_values") def map_flat_values(op, *args, **kwargs): """Applies `op` to the inner values of one or more RaggedTensors. diff --git a/tensorflow/python/ops/ragged/ragged_math_ops.py b/tensorflow/python/ops/ragged/ragged_math_ops.py index f774c1eb583619f72d23766d6534a67933673d58..02e927b6991f8d86176c347442a2f49cfdf4ce92 100644 --- a/tensorflow/python/ops/ragged/ragged_math_ops.py +++ b/tensorflow/python/ops/ragged/ragged_math_ops.py @@ -31,12 +31,14 @@ from tensorflow.python.ops.ragged import ragged_functional_ops from tensorflow.python.ops.ragged import ragged_tensor from tensorflow.python.ops.ragged import ragged_util from tensorflow.python.ops.ragged import segment_id_ops +from tensorflow.python.util.tf_export import tf_export #=============================================================================== # ragged.range #=============================================================================== # pylint: disable=redefined-builtin +@tf_export('ragged.range') def range(starts, limits=None, deltas=1, dtype=None, name=None): """Returns a `RaggedTensor` containing the specified sequences of numbers. diff --git a/tensorflow/python/ops/ragged/ragged_tensor.py b/tensorflow/python/ops/ragged/ragged_tensor.py index 567c50203af592e57168063e20787b3ed621b8c8..acf3a3841d4414c01f819d3510e0f7fb49fe426e 100644 --- a/tensorflow/python/ops/ragged/ragged_tensor.py +++ b/tensorflow/python/ops/ragged/ragged_tensor.py @@ -32,6 +32,7 @@ from tensorflow.python.ops import math_ops from tensorflow.python.ops.ragged import ragged_tensor_value from tensorflow.python.ops.ragged import ragged_util from tensorflow.python.ops.ragged import segment_id_ops +from tensorflow.python.util.tf_export import tf_export # pylint: disable=protected-access _eval_using_default_session = ops._eval_using_default_session @@ -43,6 +44,7 @@ _eval_using_default_session = ops._eval_using_default_session #=============================================================================== +@tf_export("RaggedTensor") class RaggedTensor(object): """Represents a ragged tensor (go/ragged). diff --git a/tensorflow/python/ops/ragged/ragged_tensor_value.py b/tensorflow/python/ops/ragged/ragged_tensor_value.py index bf0ac4482a1c12eb620d33471a6474d10af11875..1162487f0fa34d32833bcba8ded451a3edb23f88 100644 --- a/tensorflow/python/ops/ragged/ragged_tensor_value.py +++ b/tensorflow/python/ops/ragged/ragged_tensor_value.py @@ -20,7 +20,10 @@ from __future__ import print_function import numpy as np +from tensorflow.python.util.tf_export import tf_export + +@tf_export(v1=["ragged.RaggedTensorValue"]) class RaggedTensorValue(object): """Represents the value of a `RaggedTensor`. diff --git a/tensorflow/python/ops/ragged/segment_id_ops.py b/tensorflow/python/ops/ragged/segment_id_ops.py index fa2970c3e75af36d3f042ab23ab70c8d2cdb36ca..ee17e4d63624c91e80471eff106e0f4256139c3f 100644 --- a/tensorflow/python/ops/ragged/segment_id_ops.py +++ b/tensorflow/python/ops/ragged/segment_id_ops.py @@ -25,10 +25,12 @@ from tensorflow.python.framework import tensor_util from tensorflow.python.ops import array_ops from tensorflow.python.ops import math_ops from tensorflow.python.ops.ragged import ragged_util +from tensorflow.python.util.tf_export import tf_export # For background on "segments" and "segment ids", see: # https://www.tensorflow.org/api_guides/python/math_ops#Segmentation +@tf_export("ragged.row_splits_to_segment_ids") def row_splits_to_segment_ids(splits, name=None): """Generates the segmentation corresponding to a RaggedTensor `splits` vector. @@ -63,6 +65,7 @@ def row_splits_to_segment_ids(splits, name=None): # For background on "segments" and "segment ids", see: # https://www.tensorflow.org/api_guides/python/math_ops#Segmentation +@tf_export("ragged.segment_ids_to_row_splits") def segment_ids_to_row_splits(segment_ids, num_segments=None, name=None): """Generates the RaggedTensor `splits` vector corresponding to a segmentation. diff --git a/tensorflow/python/tools/api/generator/api_init_files.bzl b/tensorflow/python/tools/api/generator/api_init_files.bzl index 58913b32084907d037867848fe4d1a3c48f72bd2..25d0c0f75ce17add79da8f4b0db0ae627b55ecc6 100644 --- a/tensorflow/python/tools/api/generator/api_init_files.bzl +++ b/tensorflow/python/tools/api/generator/api_init_files.bzl @@ -69,6 +69,7 @@ TENSORFLOW_API_INIT_FILES = [ "nn/__init__.py", "nn/rnn_cell/__init__.py", "quantization/__init__.py", + "ragged/__init__.py", "random/__init__.py", "saved_model/__init__.py", "sets/__init__.py", diff --git a/tensorflow/python/tools/api/generator/api_init_files_v1.bzl b/tensorflow/python/tools/api/generator/api_init_files_v1.bzl index 0937f98e759c6e3fdad43972afbcc0769d7cc73a..99c8495ce5968cfc9efe66af35c42877b0c524b8 100644 --- a/tensorflow/python/tools/api/generator/api_init_files_v1.bzl +++ b/tensorflow/python/tools/api/generator/api_init_files_v1.bzl @@ -79,6 +79,7 @@ TENSORFLOW_API_INIT_FILES_V1 = [ "profiler/__init__.py", "python_io/__init__.py", "quantization/__init__.py", + "ragged/__init__.py", "random/__init__.py", "resource_loader/__init__.py", "strings/__init__.py", diff --git a/tensorflow/python/tools/api/generator/doc_srcs.py b/tensorflow/python/tools/api/generator/doc_srcs.py index abb5886deb3d9dd2e6981ee5822b0323a87eef1d..b567eead3d0c8c3023322f95402662408152ce45 100644 --- a/tensorflow/python/tools/api/generator/doc_srcs.py +++ b/tensorflow/python/tools/api/generator/doc_srcs.py @@ -54,6 +54,7 @@ _TENSORFLOW_DOC_SOURCES = { 'nn': DocSource(docstring_module_name='ops.nn_ops'), 'nn.rnn_cell': DocSource(docstring_module_name='ops.rnn_cell'), 'python_io': DocSource(docstring_module_name='lib.io.python_io'), + 'ragged': DocSource(docstring_module_name='ops.ragged'), 'resource_loader': DocSource( docstring_module_name='platform.resource_loader'), 'sets': DocSource(docstring_module_name='ops.sets'), diff --git a/tensorflow/tools/api/golden/v1/tensorflow.-ragged-tensor.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.-ragged-tensor.pbtxt new file mode 100644 index 0000000000000000000000000000000000000000..c0ed95653552f904acea1cc82bca00773ecb792c --- /dev/null +++ b/tensorflow/tools/api/golden/v1/tensorflow.-ragged-tensor.pbtxt @@ -0,0 +1,125 @@ +path: "tensorflow.RaggedTensor" +tf_class { + is_instance: "" + is_instance: "" + member { + name: "dtype" + mtype: "" + } + member { + name: "flat_values" + mtype: "" + } + member { + name: "nested_row_splits" + mtype: "" + } + member { + name: "ragged_rank" + mtype: "" + } + member { + name: "row_splits" + mtype: "" + } + member { + name: "shape" + mtype: "" + } + member { + name: "values" + mtype: "" + } + member_method { + name: "__init__" + argspec: "args=[\'self\', \'values\', \'row_splits\', \'cached_row_lengths\', \'cached_value_rowids\', \'cached_nrows\', \'internal\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'False\'], " + } + member_method { + name: "bounding_shape" + argspec: "args=[\'self\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "from_nested_row_lengths" + argspec: "args=[\'cls\', \'flat_values\', \'nested_row_lengths\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_nested_row_splits" + argspec: "args=[\'cls\', \'flat_values\', \'nested_row_splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_nested_value_rowids" + argspec: "args=[\'cls\', \'flat_values\', \'nested_value_rowids\', \'nested_nrows\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "from_row_lengths" + argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_limits" + argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_splits" + argspec: "args=[\'cls\', \'values\', \'row_splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_starts" + argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_sparse" + argspec: "args=[\'cls\', \'st_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_tensor" + argspec: "args=[\'cls\', \'tensor\', \'lengths\', \'padding\', \'ragged_rank\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'1\', \'None\'], " + } + member_method { + name: "from_value_rowids" + argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "nested_row_lengths" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "nrows" + argspec: "args=[\'self\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"\", \'None\'], " + } + member_method { + name: "row_lengths" + argspec: "args=[\'self\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\'], " + } + member_method { + name: "row_limits" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "row_starts" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "to_list" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "to_sparse" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "to_tensor" + argspec: "args=[\'self\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "value_rowids" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "with_flat_values" + argspec: "args=[\'self\', \'new_values\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "with_values" + argspec: "args=[\'self\', \'new_values\'], varargs=None, keywords=None, defaults=None" + } +} diff --git a/tensorflow/tools/api/golden/v1/tensorflow.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.pbtxt index 60ff59196b57a417badca68b87c6e3976b8a2132..5592a4c59d5c75ff0a4ad0df3f3aaff8d82b2b4f 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.pbtxt @@ -172,6 +172,10 @@ tf_module { name: "QueueBase" mtype: "" } + member { + name: "RaggedTensor" + mtype: "" + } member { name: "RandomShuffleQueue" mtype: "" @@ -516,6 +520,10 @@ tf_module { name: "quint8" mtype: "" } + member { + name: "ragged" + mtype: "" + } member { name: "random" mtype: "" diff --git a/tensorflow/tools/api/golden/v1/tensorflow.ragged.-ragged-tensor-value.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.ragged.-ragged-tensor-value.pbtxt new file mode 100644 index 0000000000000000000000000000000000000000..96c895e0a49364b37d1578ff1a1e9214a10189df --- /dev/null +++ b/tensorflow/tools/api/golden/v1/tensorflow.ragged.-ragged-tensor-value.pbtxt @@ -0,0 +1,41 @@ +path: "tensorflow.ragged.RaggedTensorValue" +tf_class { + is_instance: "" + is_instance: "" + member { + name: "dtype" + mtype: "" + } + member { + name: "flat_values" + mtype: "" + } + member { + name: "nested_row_splits" + mtype: "" + } + member { + name: "ragged_rank" + mtype: "" + } + member { + name: "row_splits" + mtype: "" + } + member { + name: "shape" + mtype: "" + } + member { + name: "values" + mtype: "" + } + member_method { + name: "__init__" + argspec: "args=[\'self\', \'values\', \'row_splits\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "to_list" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } +} diff --git a/tensorflow/tools/api/golden/v1/tensorflow.ragged.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.ragged.pbtxt new file mode 100644 index 0000000000000000000000000000000000000000..22ca7e931f3589f11b7fc5c655d633c86716b4d8 --- /dev/null +++ b/tensorflow/tools/api/golden/v1/tensorflow.ragged.pbtxt @@ -0,0 +1,31 @@ +path: "tensorflow.ragged" +tf_module { + member { + name: "RaggedTensorValue" + mtype: "" + } + member_method { + name: "constant" + argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], " + } + member_method { + name: "constant_value" + argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], " + } + member_method { + name: "map_flat_values" + argspec: "args=[\'op\'], varargs=args, keywords=kwargs, defaults=None" + } + member_method { + name: "range" + argspec: "args=[\'starts\', \'limits\', \'deltas\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'1\', \'None\', \'None\'], " + } + member_method { + name: "row_splits_to_segment_ids" + argspec: "args=[\'splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "segment_ids_to_row_splits" + argspec: "args=[\'segment_ids\', \'num_segments\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } +} diff --git a/tensorflow/tools/api/golden/v2/tensorflow.-ragged-tensor.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.-ragged-tensor.pbtxt new file mode 100644 index 0000000000000000000000000000000000000000..c0ed95653552f904acea1cc82bca00773ecb792c --- /dev/null +++ b/tensorflow/tools/api/golden/v2/tensorflow.-ragged-tensor.pbtxt @@ -0,0 +1,125 @@ +path: "tensorflow.RaggedTensor" +tf_class { + is_instance: "" + is_instance: "" + member { + name: "dtype" + mtype: "" + } + member { + name: "flat_values" + mtype: "" + } + member { + name: "nested_row_splits" + mtype: "" + } + member { + name: "ragged_rank" + mtype: "" + } + member { + name: "row_splits" + mtype: "" + } + member { + name: "shape" + mtype: "" + } + member { + name: "values" + mtype: "" + } + member_method { + name: "__init__" + argspec: "args=[\'self\', \'values\', \'row_splits\', \'cached_row_lengths\', \'cached_value_rowids\', \'cached_nrows\', \'internal\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'False\'], " + } + member_method { + name: "bounding_shape" + argspec: "args=[\'self\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "from_nested_row_lengths" + argspec: "args=[\'cls\', \'flat_values\', \'nested_row_lengths\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_nested_row_splits" + argspec: "args=[\'cls\', \'flat_values\', \'nested_row_splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_nested_value_rowids" + argspec: "args=[\'cls\', \'flat_values\', \'nested_value_rowids\', \'nested_nrows\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "from_row_lengths" + argspec: "args=[\'cls\', \'values\', \'row_lengths\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_limits" + argspec: "args=[\'cls\', \'values\', \'row_limits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_splits" + argspec: "args=[\'cls\', \'values\', \'row_splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_row_starts" + argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_sparse" + argspec: "args=[\'cls\', \'st_input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "from_tensor" + argspec: "args=[\'cls\', \'tensor\', \'lengths\', \'padding\', \'ragged_rank\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'1\', \'None\'], " + } + member_method { + name: "from_value_rowids" + argspec: "args=[\'cls\', \'values\', \'value_rowids\', \'nrows\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "nested_row_lengths" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "nrows" + argspec: "args=[\'self\', \'out_type\', \'name\'], varargs=None, keywords=None, defaults=[\"\", \'None\'], " + } + member_method { + name: "row_lengths" + argspec: "args=[\'self\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'None\'], " + } + member_method { + name: "row_limits" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "row_starts" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "to_list" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "to_sparse" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "to_tensor" + argspec: "args=[\'self\', \'default_value\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "value_rowids" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "with_flat_values" + argspec: "args=[\'self\', \'new_values\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "with_values" + argspec: "args=[\'self\', \'new_values\'], varargs=None, keywords=None, defaults=None" + } +} diff --git a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt index 0f11107dc35a78bb24e041576626beeb6588144b..5f31d274803da9e6b511bffae7abdfcb558e7cb3 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.pbtxt @@ -32,6 +32,10 @@ tf_module { name: "Operation" mtype: "" } + member { + name: "RaggedTensor" + mtype: "" + } member { name: "RegisterGradient" mtype: "" @@ -260,6 +264,10 @@ tf_module { name: "quint8" mtype: "" } + member { + name: "ragged" + mtype: "" + } member { name: "random" mtype: "" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.ragged.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.ragged.pbtxt new file mode 100644 index 0000000000000000000000000000000000000000..5fde488ffdd4dc30695407b5eba097585c885f65 --- /dev/null +++ b/tensorflow/tools/api/golden/v2/tensorflow.ragged.pbtxt @@ -0,0 +1,23 @@ +path: "tensorflow.ragged" +tf_module { + member_method { + name: "constant" + argspec: "args=[\'pylist\', \'dtype\', \'ragged_rank\', \'inner_shape\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], " + } + member_method { + name: "map_flat_values" + argspec: "args=[\'op\'], varargs=args, keywords=kwargs, defaults=None" + } + member_method { + name: "range" + argspec: "args=[\'starts\', \'limits\', \'deltas\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'1\', \'None\', \'None\'], " + } + member_method { + name: "row_splits_to_segment_ids" + argspec: "args=[\'splits\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "segment_ids_to_row_splits" + argspec: "args=[\'segment_ids\', \'num_segments\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } +} diff --git a/tensorflow/tools/compatibility/renames_v2.py b/tensorflow/tools/compatibility/renames_v2.py index 3ab5a0d0d68039d68c7387eef26e45d85bfeffc3..9a3f4460f783b0935a7623d5dbaa2f38a2c67196 100644 --- a/tensorflow/tools/compatibility/renames_v2.py +++ b/tensorflow/tools/compatibility/renames_v2.py @@ -421,6 +421,9 @@ renames = { 'tf.qr': 'tf.linalg.qr', 'tf.quantize': 'tf.quantization.quantize', 'tf.quantized_concat': 'tf.quantization.quantized_concat', + 'tf.ragged.constant_value': 'tf.compat.v1.ragged.constant_value', + 'tf.ragged.convert_to_tensor_or_ragged_tensor': 'tf.compat.v1.ragged.convert_to_tensor_or_ragged_tensor', + 'tf.ragged.RaggedTensorValue': 'tf.compat.v1.ragged.RaggedTensorValue', 'tf.random.get_seed': 'tf.compat.v1.random.get_seed', 'tf.random.set_random_seed': 'tf.compat.v1.random.set_random_seed', 'tf.random_crop': 'tf.image.random_crop',