提交 68834966 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Export ragged ops.

PiperOrigin-RevId: 224937131
上级 c07721a4
......@@ -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",
],
)
......
......@@ -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.
......
......@@ -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.
......
......@@ -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.
......
......@@ -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).
......
......@@ -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`.
......
......@@ -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.
......
......@@ -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",
......
......@@ -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",
......
......@@ -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'),
......
path: "tensorflow.RaggedTensor"
tf_class {
is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensor\'>"
is_instance: "<type \'object\'>"
member {
name: "dtype"
mtype: "<type \'property\'>"
}
member {
name: "flat_values"
mtype: "<type \'property\'>"
}
member {
name: "nested_row_splits"
mtype: "<type \'property\'>"
}
member {
name: "ragged_rank"
mtype: "<type \'property\'>"
}
member {
name: "row_splits"
mtype: "<type \'property\'>"
}
member {
name: "shape"
mtype: "<type \'property\'>"
}
member {
name: "values"
mtype: "<type \'property\'>"
}
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=[\"<dtype: \'int64\'>\", \'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"
}
}
......@@ -172,6 +172,10 @@ tf_module {
name: "QueueBase"
mtype: "<type \'type\'>"
}
member {
name: "RaggedTensor"
mtype: "<type \'type\'>"
}
member {
name: "RandomShuffleQueue"
mtype: "<type \'type\'>"
......@@ -516,6 +520,10 @@ tf_module {
name: "quint8"
mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
}
member {
name: "ragged"
mtype: "<type \'module\'>"
}
member {
name: "random"
mtype: "<type \'module\'>"
......
path: "tensorflow.ragged.RaggedTensorValue"
tf_class {
is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor_value.RaggedTensorValue\'>"
is_instance: "<type \'object\'>"
member {
name: "dtype"
mtype: "<type \'property\'>"
}
member {
name: "flat_values"
mtype: "<type \'property\'>"
}
member {
name: "nested_row_splits"
mtype: "<type \'property\'>"
}
member {
name: "ragged_rank"
mtype: "<type \'property\'>"
}
member {
name: "row_splits"
mtype: "<type \'property\'>"
}
member {
name: "shape"
mtype: "<type \'property\'>"
}
member {
name: "values"
mtype: "<type \'property\'>"
}
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"
}
}
path: "tensorflow.ragged"
tf_module {
member {
name: "RaggedTensorValue"
mtype: "<type \'type\'>"
}
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\'], "
}
}
path: "tensorflow.RaggedTensor"
tf_class {
is_instance: "<class \'tensorflow.python.ops.ragged.ragged_tensor.RaggedTensor\'>"
is_instance: "<type \'object\'>"
member {
name: "dtype"
mtype: "<type \'property\'>"
}
member {
name: "flat_values"
mtype: "<type \'property\'>"
}
member {
name: "nested_row_splits"
mtype: "<type \'property\'>"
}
member {
name: "ragged_rank"
mtype: "<type \'property\'>"
}
member {
name: "row_splits"
mtype: "<type \'property\'>"
}
member {
name: "shape"
mtype: "<type \'property\'>"
}
member {
name: "values"
mtype: "<type \'property\'>"
}
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=[\"<dtype: \'int64\'>\", \'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"
}
}
......@@ -32,6 +32,10 @@ tf_module {
name: "Operation"
mtype: "<type \'type\'>"
}
member {
name: "RaggedTensor"
mtype: "<type \'type\'>"
}
member {
name: "RegisterGradient"
mtype: "<type \'type\'>"
......@@ -260,6 +264,10 @@ tf_module {
name: "quint8"
mtype: "<class \'tensorflow.python.framework.dtypes.DType\'>"
}
member {
name: "ragged"
mtype: "<type \'module\'>"
}
member {
name: "random"
mtype: "<type \'module\'>"
......
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\'], "
}
}
......@@ -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',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册