tensorflow--tensorflow/tensorflow/python/data/experimental/ops/BUILD

load("@xla//third_party/rules_python/python:py_library.bzl", "py_library")
load("//tensorflow:tensorflow.default.bzl", "tf_py_strict_test")

package(
    # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
    default_visibility = ["//tensorflow:internal"],
    licenses = ["notice"],
)

py_library(
    name = "batching",
    srcs = ["batching.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:structured_function",
        "//tensorflow/python/data/util:convert",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:sparse_tensor",
        "//tensorflow/python/framework:tensor_shape",
        "//tensorflow/python/framework:tensor_util",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "cardinality",
    srcs = ["cardinality.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:dataset_ops_gen",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "compression_ops",
    srcs = ["compression_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "counter",
    srcs = ["counter.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python:tf2",
        "//tensorflow/python/compat:v2_compat",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "data_service_ops",
    srcs = [
        "data_service_ops.py",
    ],
    lazy_imports = True,
    strict_deps = True,
    deps = [
        ":compression_ops",
        "//tensorflow/core/protobuf:for_core_protos_py_proto",
        "//tensorflow/python:tf2",
        "//tensorflow/python/data/experimental/service:_pywrap_server_lib",
        "//tensorflow/python/data/experimental/service:_pywrap_utils_exp",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:options",
        "//tensorflow/python/data/ops:structured_function",
        "//tensorflow/python/eager:context",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor",
        "//tensorflow/python/framework:tensor_util",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/ops:string_ops",
        "//tensorflow/python/saved_model:nested_structure_coder",
        "//tensorflow/python/util:nest",
        "//tensorflow/python/util:tf_export",
    ],
)

tf_py_strict_test(
    name = "data_service_ops_test",
    srcs = ["data_service_ops_test.py"],
    deps = [
        ":data_service_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:tensor_shape",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops/ragged:ragged_tensor",
        "//tensorflow/python/platform:client_testlib",
    ],
)

py_library(
    name = "distributed_save_op",
    srcs = [
        "distributed_save_op.py",
    ],
    strict_deps = True,
    visibility = ["//tensorflow:internal"],
    deps = [
        "//tensorflow/core:protos_all_py",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/saved_model:nested_structure_coder",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "distribute",
    srcs = [
        "distribute.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:options",
        "//tensorflow/python/data/util:nest",
        "//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:experimental_dataset_ops_gen",
        "//tensorflow/python/types:data",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "enumerate_ops",
    srcs = ["enumerate_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "error_ops",
    srcs = ["error_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "from_list",
    srcs = ["from_list.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "get_single_element",
    srcs = ["get_single_element.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/types:data",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "global_shuffle_op",
    srcs = [
        "global_shuffle_op.py",
    ],
    strict_deps = True,
    visibility = ["//tensorflow:internal"],
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:options",
        "//tensorflow/python/data/util:random_seed",
        "//tensorflow/python/framework:tensor",
        "//tensorflow/python/ops:dataset_ops_gen",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "grouping",
    srcs = ["grouping.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:structured_function",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "index_flat_map_op",
    srcs = [
        "index_flat_map_op.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:structured_function",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "interleave_ops",
    srcs = ["interleave_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python:tf2",
        "//tensorflow/python/compat:v2_compat",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:readers",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "io",
    srcs = [
        "io.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "iterator_ops",
    srcs = [
        "iterator_ops.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:iterator_ops",
        "//tensorflow/python/data/ops:options",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:proto_ops",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "iterator_model_ops",
    srcs = [
        "iterator_model_ops.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/core:protos_all_py",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:iterator_ops",
        "//tensorflow/python/eager:context",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "lookup_ops",
    srcs = [
        "lookup_ops.py",
    ],
    strict_deps = True,
    deps = [
        ":cardinality",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/ops:lookup_ops",
        "//tensorflow/python/ops:math_ops",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "map_defun",
    srcs = ["map_defun.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_shape",
        "//tensorflow/python/ops:dataset_ops_gen",
    ],
)

py_library(
    name = "matching_files",
    srcs = ["matching_files.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "pad_to_cardinality",
    srcs = ["pad_to_cardinality.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/eager:context",
        "//tensorflow/python/ops:array_ops",
        "//tensorflow/python/util:nest",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "parsing_ops",
    srcs = ["parsing_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:sparse_tensor",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/ops:parsing_ops",
        "//tensorflow/python/ops/ragged:ragged_tensor",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "prefetching_ops",
    srcs = ["prefetching_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:iterator_ops",
        "//tensorflow/python/data/ops:structured_function",
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/eager:def_function",
        "//tensorflow/python/eager:function",
        "//tensorflow/python/framework:device",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:array_ops",
        "//tensorflow/python/ops:dataset_ops_gen",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/ops:functional_ops",
        "//tensorflow/python/ops:resource_variable_ops",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "random_access",
    srcs = ["random_access.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/util:structure",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "random_ops",
    srcs = [
        "random_ops.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python:tf2",
        "//tensorflow/python/compat:v2_compat",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "readers",
    srcs = [
        "readers.py",
    ],
    strict_deps = True,
    deps = [
        ":error_ops",
        ":parsing_ops",
        "//tensorflow/python:tf2",
        "//tensorflow/python/compat:v2_compat",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:options",
        "//tensorflow/python/data/ops:readers",
        "//tensorflow/python/data/util:convert",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/framework:constant_op",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/framework:tensor_util",
        "//tensorflow/python/lib/io:file_io",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/ops:io_ops",
        "//tensorflow/python/platform:gfile",
        "//tensorflow/python/util:tf_export",
        "//third_party/py/numpy",
    ],
)

py_library(
    name = "resampling",
    srcs = ["resampling.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "scan_ops",
    srcs = ["scan_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "shuffle_ops",
    srcs = [
        "shuffle_ops.py",
    ],
    strict_deps = True,
    visibility = [
        # TODO(jsimsa): Remove non-default visibility exception when `index_shuffle` is made public.
        "//third_party/py/tensorflow_datasets:__subpackages__",
        "//tensorflow:internal",
    ],
    deps = [
        ":random_access",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/util:random_seed",
        "//tensorflow/python/framework:constant_op",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:array_ops",
        "//tensorflow/python/ops:dataset_ops_gen",
        "//tensorflow/python/ops:math_ops",
        "//tensorflow/python/ops:stateless_random_ops",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
        "//third_party/py/numpy",
    ],
)

py_library(
    name = "snapshot",
    srcs = [
        "snapshot.py",
    ],
    strict_deps = True,
    visibility = ["//tensorflow:internal"],
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:random_seed",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "take_while_ops",
    srcs = ["take_while_ops.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "testing",
    srcs = ["testing.py"],
    strict_deps = True,
    deps = [
        "//tensorflow/core:protos_all_py",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "unique",
    srcs = [
        "unique.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "weighted_flat_map_op",
    srcs = [
        "weighted_flat_map_op.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/util:convert",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/data/util:random_seed",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor",
        "//tensorflow/python/ops:dataset_ops_gen",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
    ],
)

py_library(
    name = "writers",
    srcs = [
        "writers.py",
    ],
    strict_deps = True,
    deps = [
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/util:convert",
        "//tensorflow/python/framework:dtypes",
        "//tensorflow/python/framework:ops",
        "//tensorflow/python/framework:tensor_spec",
        "//tensorflow/python/ops:experimental_dataset_ops_gen",
        "//tensorflow/python/types:data",
        "//tensorflow/python/util:deprecation",
        "//tensorflow/python/util:tf_export",
    ],
)

py_library(
    name = "dataset_ops",
    strict_deps = True,
    deps = [
        ":batching",
        ":cardinality",
        ":compression_ops",
        ":counter",
        ":data_service_ops",
        ":distribute",
        ":distributed_save_op",
        ":enumerate_ops",
        ":error_ops",
        ":from_list",
        ":get_single_element",
        ":global_shuffle_op",
        ":grouping",
        ":interleave_ops",
        ":io",
        ":map_defun",
        ":matching_files",
        ":pad_to_cardinality",
        ":prefetching_ops",
        ":random_access",
        ":readers",
        ":resampling",
        ":scan_ops",
        ":shuffle_ops",
        ":snapshot",
        ":take_while_ops",
        ":unique",
        ":writers",
        "//tensorflow/python/data/ops:dataset_ops",
        "//tensorflow/python/data/ops:debug_mode",
        "//tensorflow/python/data/util:nest",
        "//tensorflow/python/ops:dataset_ops_gen",
    ],
)