wehub-resource-sync
1800 lines
57 KiB
Python
1800 lines
57 KiB
Python
load("@xla//third_party/rules_python/python:py_library.bzl", "py_library")
|
|
load("//tensorflow:tensorflow.bzl", "py_test")
|
|
load("//tensorflow:tensorflow.default.bzl", "cuda_py_strict_test")
|
|
load("//tensorflow/tools/test:performance.bzl", "tf_py_logged_benchmark")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
|
|
default_visibility = ["//tensorflow:internal"],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
#-------------------------------------------------------------------------------
|
|
# RaggedTensor
|
|
#-------------------------------------------------------------------------------
|
|
|
|
py_library(
|
|
name = "ragged",
|
|
srcs = ["__init__.py"],
|
|
strict_deps = True,
|
|
tags = ["nofixdeps"],
|
|
deps = [
|
|
":dynamic_ragged_shape",
|
|
":ragged_array_ops",
|
|
":ragged_batch_gather_ops",
|
|
":ragged_batch_gather_with_default_op",
|
|
":ragged_check_ops",
|
|
":ragged_concat_ops",
|
|
":ragged_config",
|
|
":ragged_conversion_ops",
|
|
":ragged_dispatch",
|
|
":ragged_embedding_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_gather_ops",
|
|
":ragged_getitem",
|
|
":ragged_image_ops",
|
|
":ragged_map_ops",
|
|
":ragged_math_ops",
|
|
":ragged_operators",
|
|
":ragged_ops",
|
|
":ragged_string_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
":ragged_tensor_value",
|
|
":ragged_util",
|
|
":ragged_where_op",
|
|
":segment_id_ops",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_array_ops",
|
|
srcs = ["ragged_array_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":dynamic_ragged_shape",
|
|
":ragged_functional_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
":ragged_util",
|
|
":segment_id_ops",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:data_flow_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_array_ops_gen",
|
|
"//tensorflow/python/ops:sort_ops",
|
|
"//tensorflow/python/types:core",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_autograph",
|
|
srcs = ["ragged_autograph.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/autograph/operators:control_flow",
|
|
"//tensorflow/python/ops:cond",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_check_ops",
|
|
srcs = ["ragged_check_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_batch_gather_ops",
|
|
srcs = ["ragged_batch_gather_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_gather_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_batch_gather_with_default_op",
|
|
srcs = [
|
|
"ragged_batch_gather_with_default_op.py",
|
|
],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_dispatch",
|
|
":ragged_operators",
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
":ragged_where_op",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_bincount_ops",
|
|
srcs = ["ragged_bincount_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:bincount_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:count_ops_gen",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:math_ops_gen",
|
|
"//tensorflow/python/ops:sparse_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
cuda_py_strict_test(
|
|
name = "ragged_bincount_ops_test",
|
|
size = "small",
|
|
srcs = ["ragged_bincount_ops_test.py"],
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:config",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:bincount_ops",
|
|
"//tensorflow/python/ops:sparse_ops",
|
|
"//tensorflow/python/platform:client_testlib",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_concat_ops",
|
|
srcs = ["ragged_concat_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_gather_ops",
|
|
":ragged_tensor",
|
|
":ragged_util",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_conversion_ops",
|
|
srcs = ["ragged_conversion_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:indexed_slices",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_conversion_ops_gen",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_embedding_ops",
|
|
srcs = ["ragged_embedding_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:embedding_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:resource_variable_ops",
|
|
"//tensorflow/python/ops:variables",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_factory_ops",
|
|
srcs = ["ragged_factory_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
":ragged_tensor_value",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:numpy_compat",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_functional_ops",
|
|
srcs = ["ragged_functional_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_config",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_gather_ops",
|
|
srcs = ["ragged_gather_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:indexed_slices",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_array_ops_gen",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_getitem",
|
|
srcs = ["ragged_getitem.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_gather_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_math_ops",
|
|
srcs = ["ragged_math_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_functional_ops",
|
|
":ragged_tensor",
|
|
":segment_id_ops",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/ops:ragged_math_ops_gen",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_operators",
|
|
srcs = ["ragged_operators.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_getitem",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:tf_decorator_py",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_ops",
|
|
srcs = ["ragged_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_autograph",
|
|
":ragged_batch_gather_ops",
|
|
":ragged_batch_gather_with_default_op",
|
|
":ragged_bincount_ops",
|
|
":ragged_check_ops",
|
|
":ragged_concat_ops",
|
|
":ragged_config", # fixdeps: keep
|
|
":ragged_conversion_ops",
|
|
":ragged_dispatch",
|
|
":ragged_embedding_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_gather_ops",
|
|
":ragged_getitem",
|
|
":ragged_image_ops",
|
|
":ragged_map_ops",
|
|
":ragged_math_ops",
|
|
":ragged_operators",
|
|
":ragged_squeeze_op",
|
|
":ragged_string_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
":ragged_tensor_value",
|
|
":ragged_where_op",
|
|
":segment_id_ops",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_string_ops",
|
|
srcs = ["ragged_string_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/ops:string_ops_gen",
|
|
"//tensorflow/python/util:compat",
|
|
"//tensorflow/python/util:deprecation",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_squeeze_op",
|
|
srcs = ["ragged_squeeze_op.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:control_flow_assert",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:deprecation",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_config",
|
|
srcs = ["ragged_config.py"],
|
|
strict_deps = True,
|
|
)
|
|
|
|
py_library(
|
|
name = "row_partition",
|
|
srcs = ["row_partition.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":segment_id_ops",
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python/framework:composite_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_conversion",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/framework:type_spec",
|
|
"//tensorflow/python/framework:type_spec_registry",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:bincount_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_math_ops_gen",
|
|
"//tensorflow/python/saved_model:nested_structure_coder",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_tensor",
|
|
# RaggedTensor must import ragged_ops to ensure that all dispatched
|
|
# ragged ops are registered.
|
|
# Build cycles from this multi-src build target should not occur
|
|
# if proper TF layering is followed, ie ragged_tensor is allowed to depend on
|
|
# other ops but other ops are not allowed to depend on ragged_tensor.
|
|
srcs = [
|
|
"dynamic_ragged_shape.py",
|
|
"ragged_array_ops.py",
|
|
"ragged_autograph.py",
|
|
"ragged_batch_gather_ops.py",
|
|
"ragged_batch_gather_with_default_op.py",
|
|
"ragged_bincount_ops.py",
|
|
"ragged_check_ops.py",
|
|
"ragged_concat_ops.py",
|
|
"ragged_conversion_ops.py",
|
|
"ragged_dispatch.py",
|
|
"ragged_embedding_ops.py",
|
|
"ragged_factory_ops.py",
|
|
"ragged_functional_ops.py",
|
|
"ragged_gather_ops.py",
|
|
"ragged_getitem.py",
|
|
"ragged_image_ops.py",
|
|
"ragged_map_ops.py",
|
|
"ragged_math_ops.py",
|
|
"ragged_operators.py",
|
|
"ragged_ops.py",
|
|
"ragged_squeeze_op.py",
|
|
"ragged_string_ops.py",
|
|
"ragged_tensor.py",
|
|
"ragged_tensor_shape.py",
|
|
"ragged_where_op.py",
|
|
],
|
|
strict_deps = True,
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
":ragged_config",
|
|
":ragged_tensor_value",
|
|
":ragged_util",
|
|
":row_partition",
|
|
":segment_id_ops",
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python:tf2",
|
|
"//tensorflow/python/autograph/operators:control_flow",
|
|
"//tensorflow/python/client:session",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:composite_tensor",
|
|
"//tensorflow/python/framework:composite_tensor_gradient",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:extension_type",
|
|
"//tensorflow/python/framework:indexed_slices",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_conversion",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/framework:type_spec",
|
|
"//tensorflow/python/framework:type_spec_registry",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_gen",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:bincount_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:control_flow_assert",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:count_ops_gen",
|
|
"//tensorflow/python/ops:data_flow_ops",
|
|
"//tensorflow/python/ops:embedding_ops",
|
|
"//tensorflow/python/ops:image_ops",
|
|
"//tensorflow/python/ops:logging_ops",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:math_ops_gen",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/ops:ragged_array_ops_gen",
|
|
"//tensorflow/python/ops:ragged_conversion_ops_gen",
|
|
"//tensorflow/python/ops:ragged_math_ops_gen",
|
|
"//tensorflow/python/ops:sort_ops",
|
|
"//tensorflow/python/ops:sparse_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/ops:string_ops_gen",
|
|
"//tensorflow/python/ops:variables",
|
|
"//tensorflow/python/saved_model:nested_structure_coder",
|
|
"//tensorflow/python/types:core",
|
|
"//tensorflow/python/types:internal",
|
|
"//tensorflow/python/util:compat",
|
|
"//tensorflow/python/util:deprecation",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:nest",
|
|
"//tensorflow/python/util:numpy_compat",
|
|
"//tensorflow/python/util:tf_decorator_py",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//tensorflow/python/util:tf_inspect",
|
|
"//tensorflow/tools/docs:doc_controls",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_tensor_shape",
|
|
srcs = ["ragged_tensor_shape.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_config",
|
|
":ragged_tensor",
|
|
":ragged_util",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:control_flow_assert",
|
|
"//tensorflow/python/ops:math_ops",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "dynamic_ragged_shape",
|
|
srcs = ["dynamic_ragged_shape.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
":row_partition",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:extension_type",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/types:core",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_tensor_value",
|
|
srcs = ["ragged_tensor_value.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":row_partition",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_util",
|
|
srcs = ["ragged_util.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:control_flow_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_math_ops_gen",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_where_op",
|
|
srcs = ["ragged_where_op.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_concat_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_gather_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "segment_id_ops",
|
|
srcs = ["segment_id_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_util",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:bincount_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_export",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_image_ops",
|
|
srcs = ["ragged_image_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:image_ops",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/util:dispatch",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_map_ops",
|
|
srcs = ["ragged_map_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/util:nest",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_dispatch",
|
|
srcs = ["ragged_dispatch.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":dynamic_ragged_shape", # fixdeps: keep
|
|
":ragged_array_ops", # fixdeps: keep
|
|
":ragged_batch_gather_ops", # fixdeps: keep
|
|
":ragged_concat_ops", # fixdeps: keep
|
|
":ragged_gather_ops", # fixdeps: keep
|
|
":ragged_math_ops", # fixdeps: keep
|
|
":ragged_squeeze_op", # fixdeps: keep
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
":ragged_util", # fixdeps: keep
|
|
":ragged_where_op", # fixdeps: keep
|
|
"//tensorflow/python/framework:dtypes", # fixdeps: keep
|
|
"//tensorflow/python/framework:ops", # fixdeps: keep
|
|
"//tensorflow/python/framework:sparse_tensor", # fixdeps: keep
|
|
"//tensorflow/python/ops:array_ops", # fixdeps: keep
|
|
"//tensorflow/python/ops:bitwise_ops", # fixdeps: keep
|
|
"//tensorflow/python/ops:clip_ops", # fixdeps: keep
|
|
"//tensorflow/python/ops:logging_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:parsing_ops", # fixdeps: keep
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/ops:variables", # fixdeps: keep
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:tf_decorator_py",
|
|
"//tensorflow/python/util:tf_export",
|
|
"//tensorflow/python/util:tf_inspect",
|
|
"//third_party/py/numpy", # fixdeps: keep
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "ragged_tensor_test_ops",
|
|
srcs = ["ragged_tensor_test_ops.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:bitwise_ops_gen",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:nn_impl",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/ops:parsing_ops",
|
|
"//tensorflow/python/ops:special_math_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
],
|
|
)
|
|
|
|
#-------------------------------------------------------------------------------
|
|
# RaggedTensor Tests
|
|
#-------------------------------------------------------------------------------
|
|
|
|
py_test(
|
|
name = "ragged_tensor_test",
|
|
srcs = ["ragged_tensor_test.py"],
|
|
shard_count = 8,
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_concat_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_gather_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_value",
|
|
":row_partition",
|
|
"//tensorflow/core:protos_all_py",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/eager:backprop",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/framework:type_spec",
|
|
"//tensorflow/python/framework:type_utils",
|
|
"//tensorflow/python/ops:array_grad",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:cond",
|
|
"//tensorflow/python/ops:gradients_impl",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_grad",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:ragged_conversion_ops_gen",
|
|
"//tensorflow/python/ops:tensor_array_grad",
|
|
"//tensorflow/python/platform:test",
|
|
"//tensorflow/python/util:nest",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_math_ops_test",
|
|
size = "medium",
|
|
srcs = ["ragged_math_ops_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_math_ops",
|
|
":ragged_string_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_getitem_test",
|
|
size = "medium",
|
|
timeout = "long",
|
|
srcs = ["ragged_getitem_test.py"],
|
|
shard_count = 4,
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "row_partition_test",
|
|
srcs = ["row_partition_test.py"],
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":row_partition",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_eager_test",
|
|
size = "medium",
|
|
srcs = ["ragged_eager_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_range_op_test",
|
|
srcs = ["ragged_range_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_math_ops",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:ragged_math_ops_gen",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_tensor_bounding_shape_op_test",
|
|
srcs = ["ragged_tensor_bounding_shape_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_row_lengths_op_test",
|
|
srcs = ["ragged_row_lengths_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_gather_op_test",
|
|
srcs = ["ragged_gather_op_test.py"],
|
|
shard_count = 4,
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_gather_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:indexed_slices",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:gradients_impl",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_batch_gather_op_test",
|
|
srcs = ["ragged_batch_gather_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_batch_gather_ops",
|
|
":ragged_batch_gather_with_default_op",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_gather_nd_op_test",
|
|
srcs = ["ragged_gather_nd_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_gather_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_row_splits_to_segment_ids_op_test",
|
|
srcs = ["ragged_row_splits_to_segment_ids_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":segment_id_ops",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_segment_ids_to_row_splits_op_test",
|
|
srcs = ["ragged_segment_ids_to_row_splits_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":segment_id_ops",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_from_tensor_op_test",
|
|
srcs = ["ragged_from_tensor_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_to_sparse_op_test",
|
|
srcs = ["ragged_to_sparse_op_test.py"],
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:gradients_impl",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_from_sparse_op_test",
|
|
srcs = ["ragged_from_sparse_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_to_tensor_op_test",
|
|
srcs = ["ragged_to_tensor_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/client:session",
|
|
"//tensorflow/python/eager:backprop",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:gradients_impl",
|
|
"//tensorflow/python/platform:benchmark",
|
|
"//tensorflow/python/platform:test",
|
|
"//tensorflow/python/util:nest",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
tf_py_logged_benchmark(
|
|
name = "ragged_to_tensor_op_benchmark",
|
|
target = "//tensorflow/python/ops/ragged:ragged_to_tensor_op_test",
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_segment_op_test",
|
|
srcs = ["ragged_segment_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_reduce_op_test",
|
|
srcs = ["ragged_reduce_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_math_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_map_flat_values_op_test",
|
|
srcs = ["ragged_map_flat_values_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_const_op_test",
|
|
srcs = ["ragged_const_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "strings_reduce_join_op_test",
|
|
srcs = ["strings_reduce_join_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_string_ops",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_constant_value_op_test",
|
|
srcs = ["ragged_constant_value_op_test.py"],
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows",
|
|
],
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor_value",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "convert_to_tensor_or_ragged_tensor_op_test",
|
|
srcs = ["convert_to_tensor_or_ragged_tensor_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_boolean_mask_op_test",
|
|
srcs = ["ragged_boolean_mask_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_concat_op_test",
|
|
srcs = ["ragged_concat_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_concat_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_stack_op_test",
|
|
srcs = ["ragged_stack_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_concat_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_rank_op_test",
|
|
srcs = ["ragged_rank_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_tile_op_test",
|
|
srcs = ["ragged_tile_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_util_test",
|
|
srcs = ["ragged_util_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_util",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_expand_dims_op_test",
|
|
srcs = ["ragged_expand_dims_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_where_op_test",
|
|
srcs = ["ragged_where_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_where_op",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_dispatch_test",
|
|
srcs = ["ragged_dispatch_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":dynamic_ragged_shape",
|
|
":ragged", # fixdeps: keep
|
|
":ragged_dispatch",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_test_ops",
|
|
"//tensorflow/python:tf2",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:random_seed",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:clip_ops",
|
|
"//tensorflow/python/ops:data_flow_ops",
|
|
"//tensorflow/python/ops:image_ops_impl",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:nn_impl",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/ops:variables",
|
|
"//tensorflow/python/platform:test",
|
|
"//tensorflow/python/util:dispatch",
|
|
"//tensorflow/python/util:nest",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_operators_test",
|
|
srcs = ["ragged_operators_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python:tf2",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_resize_image_op_test",
|
|
srcs = ["ragged_resize_image_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_concat_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:image_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_map_fn_op_test",
|
|
size = "small",
|
|
srcs = ["ragged_map_fn_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_functional_ops",
|
|
":ragged_map_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_tensor_shape_test",
|
|
srcs = ["ragged_tensor_shape_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_shape",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "dynamic_ragged_shape_test",
|
|
size = "medium",
|
|
srcs = ["dynamic_ragged_shape_test.py"],
|
|
shard_count = 8,
|
|
strict_deps = True,
|
|
deps = [
|
|
":dynamic_ragged_shape",
|
|
":ragged", # fixdeps: keep
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
":row_partition",
|
|
"//tensorflow/python/client:session",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:tensor_util",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:array_ops_stack",
|
|
"//tensorflow/python/ops:gradients_impl",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:math_ops_gen",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_size_op_test",
|
|
srcs = ["ragged_size_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_placeholder_op_test",
|
|
srcs = ["ragged_placeholder_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_squeeze_op_test",
|
|
srcs = ["ragged_squeeze_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_conversion_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_squeeze_op",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_strict_test(
|
|
name = "ragged_dynamic_partition_op_test",
|
|
srcs = ["ragged_dynamic_partition_op_test.py"],
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_concat_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:data_flow_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_merge_dims_op_test",
|
|
srcs = ["ragged_merge_dims_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//tensorflow/python/util:nest",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "string_ngrams_op_test",
|
|
size = "small",
|
|
srcs = ["string_ngrams_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_string_ops",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:client_testlib",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_reverse_op_test",
|
|
srcs = ["ragged_reverse_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_cross_op_test",
|
|
srcs = ["ragged_cross_op_test.py"],
|
|
strict_deps = True,
|
|
tags = [
|
|
"no_windows", # TODO(b/150702952): Reenable windows test once fixed.
|
|
],
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:sparse_tensor",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:ragged_array_ops_gen",
|
|
"//tensorflow/python/ops:random_ops",
|
|
"//tensorflow/python/ops:sparse_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_one_hot_op_test",
|
|
srcs = ["ragged_one_hot_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_print_op_test",
|
|
srcs = ["ragged_print_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged", # fixdeps: keep
|
|
":ragged_factory_ops",
|
|
":ragged_string_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:logging_ops",
|
|
"//tensorflow/python/ops:sparse_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_tensor_supported_values_test",
|
|
srcs = ["ragged_tensor_supported_values_test.py"],
|
|
shard_count = 10,
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
":ragged_tensor_test_ops",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:extension_type",
|
|
"//tensorflow/python/framework:tensor",
|
|
"//tensorflow/python/framework:tensor_shape",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/framework:type_spec",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:check_ops",
|
|
"//tensorflow/python/ops:clip_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:nn_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"//tensorflow/python/util:dispatch",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_matmul_op_test",
|
|
srcs = ["ragged_matmul_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_concat_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_math_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:array_ops",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_split_op_test",
|
|
srcs = ["ragged_split_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:errors",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:tensor_spec",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
cuda_py_strict_test(
|
|
name = "ragged_factory_ops_test",
|
|
srcs = ["ragged_factory_ops_test.py"],
|
|
deps = [
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/data/ops:dataset_ops",
|
|
"//tensorflow/python/data/ops:multi_device_iterator_ops",
|
|
"//tensorflow/python/distribute:mirrored_strategy",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:ops",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:map_fn",
|
|
"//tensorflow/python/ops:math_ops",
|
|
"//tensorflow/python/ops:string_ops",
|
|
"//tensorflow/python/platform:client_testlib",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_bitcast_op_test",
|
|
srcs = ["ragged_bitcast_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/platform:test",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|
|
|
|
py_test(
|
|
name = "ragged_fill_empty_rows_op_test",
|
|
srcs = ["ragged_fill_empty_rows_op_test.py"],
|
|
strict_deps = True,
|
|
deps = [
|
|
":ragged_array_ops",
|
|
":ragged_factory_ops",
|
|
":ragged_tensor",
|
|
"//tensorflow/python/eager:def_function",
|
|
"//tensorflow/python/framework:constant_op",
|
|
"//tensorflow/python/framework:dtypes",
|
|
"//tensorflow/python/framework:test_lib",
|
|
"//tensorflow/python/ops:gradient_checker",
|
|
"//tensorflow/python/platform:test",
|
|
"//third_party/py/numpy",
|
|
],
|
|
)
|