| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 1 [//tensorflow/python/util:cpp_python_util] # util tfe
2 tensorflow::swig::IsSequence
3 tensorflow::swig::IsSequenceOrComposite
4 tensorflow::swig::IsCompositeTensor
5 tensorflow::swig::IsTypeSpec
6 tensorflow::swig::IsNamedtuple
7 tensorflow::swig::IsMapping
8 tensorflow::swig::IsMutableMapping
9 tensorflow::swig::IsMappingView
10 tensorflow::swig::IsAttrs
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | BUILD | 4 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5 load("//tensorflow:tensorflow.bzl", "tf_cc_test")
8 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
9 load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # buildifier: disable=same-origin-load
12 default_visibility = ["//tensorflow:internal"],
49 "//tensorflow/core:dataset_ops_op_lib",
50 "//tensorflow/core:framework",
51 "//tensorflow/core:lib",
52 "//tensorflow/core:lib_internal",
53 "//tensorflow/core/data:dataset_utils",
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | BUILD | 2 load("//tensorflow:tensorflow.bzl", "cuda_py_test") # buildifier: disable=same-origin-load
3 load("//tensorflow:tensorflow.bzl", "tf_py_test") # buildifier: disable=same-origin-load
6 default_visibility = ["//tensorflow:internal"],
16 "//tensorflow/python:client_testlib",
17 "//tensorflow/python:constant_op",
18 "//tensorflow/python:framework_combinations",
19 "//tensorflow/python:sparse_tensor",
20 "//tensorflow/python/data/ops:dataset_ops",
21 "//tensorflow/python/ops/ragged:ragged_tensor_value",
35 "//tensorflow/python:array_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "tf_custom_op_library")
4 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
7 load("//tensorflow:tensorflow.bzl", "tf_py_test")
10 default_visibility = ["//tensorflow:internal"],
24 "//tensorflow/python:array_ops",
25 "//tensorflow/python:client_testlib",
26 "//tensorflow/python:framework_for_generated_wrappers",
27 "//tensorflow/python:string_ops",
37 "//tensorflow/python:array_ops",
38 "//tensorflow/python:client_testlib",
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | BUILD | 1 load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library")
2 load("//tensorflow:tensorflow.bzl", "tf_py_test")
3 load("//tensorflow/core/platform/default:distribute.bzl", "distribute_py_test")
6 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
9 load("//tensorflow:tensorflow.bzl", "cuda_py_tests")
12 default_visibility = ["//tensorflow:internal"],
17 # Used in a pybind extension whose rule must be in tensorflow/python
19 visibility = ["//tensorflow/python:__pkg__"],
23 # Used in a rule which visibility limits to tensorflow/python
25 visibility = ["//tensorflow/python:__pkg__"],
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2 load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cc_test_mkl", "tf_copts", "tf_cuda_cc_test")
6 load("//tensorflow:tensorflow.bzl", "filegroup")
9 load("//tensorflow:tensorflow.bzl", "tf_kernel_library")
13 "//tensorflow/core/platform:build_config_root.bzl",
29 "//tensorflow/python/grappler:__pkg__",
41 "//tensorflow/core:lib",
42 "//tensorflow/core:protos_all_cc",
43 "//tensorflow/core/grappler:grappler_item",
44 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | BUILD | 5 "//tensorflow:tensorflow.bzl",
9 load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # buildifier: disable=same-origin-load
10 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
13 default_visibility = ["//tensorflow:internal"],
33 visibility = ["//tensorflow:__subpackages__"],
41 "//tensorflow/core:experimental_dataset_ops_op_lib",
42 "//tensorflow/core:framework",
43 "//tensorflow/core/data:name_utils",
52 "//tensorflow/core:experimental_dataset_ops_op_lib",
53 "//tensorflow/core:framework",
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | BUILD | 4 load("//tensorflow:tensorflow.bzl", "py_strict_library")
5 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
8 load("//tensorflow:tensorflow.bzl", "tf_py_test")
11 load("//tensorflow:tensorflow.bzl", "tf_python_pybind_extension")
14 load("//tensorflow:tensorflow.bzl", "tf_pybind_cc_library_wrapper")
17 # TODO(drpng): change that to //third_party/tensorflow:internal
27 visibility = ["//tensorflow:internal"],
41 "//tensorflow/python:util",
42 "//tensorflow/python/saved_model/model_utils",
52 "//tensorflow/python/util:tf_export",
[all …]
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 1 tensorflow/__init__.py
2 tensorflow/api_template.__init__.py
3 tensorflow/api_template_v1.__init__.py
4 tensorflow/compat_template.__init__.py
5 tensorflow/compat_template_v1.__init__.py
6 tensorflow/compiler/mlir/glob_lit_test.bzl
7 tensorflow/compiler/mlir/hlo/WORKSPACE
8 tensorflow/compiler/mlir/tfrt/transforms/tpu_passes.h
9 tensorflow/core/tfrt/tpu/tpu_resources.h
10 tensorflow/go/op/wrappers.go
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "py_test", "tf_py_test")
2 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
3 load("//tensorflow/core/platform/default:distribute.bzl", "distribute_py_test")
4 load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
8 "//tensorflow:internal",
25 "//tensorflow/python/distribute/v1:all_reduce",
43 "//tensorflow/python:array_ops",
44 "//tensorflow/python:framework_ops",
45 "//tensorflow/python:kernels",
46 "//tensorflow/python:math_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | BUILD | 2 load("//tensorflow:tensorflow.bzl", "cuda_py_test") # buildifier: disable=same-origin-load
3 load("//tensorflow:tensorflow.bzl", "tf_py_test") # buildifier: disable=same-origin-load
6 default_visibility = ["//tensorflow:internal"],
15 "//tensorflow/python/data/experimental/ops:cardinality",
16 "//tensorflow/python/data/kernel_tests:checkpoint_test_base",
17 "//tensorflow/python/data/kernel_tests:test_base",
18 "//tensorflow/python/data/ops:dataset_ops",
28 "//tensorflow/python:client_testlib",
29 "//tensorflow/python:errors",
30 "//tensorflow/python/data/experimental/ops:testing",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 4 load("//tensorflow/compiler/xla/tests:build_defs.bzl", "generate_backend_suites", "generate_backend…
6 "//tensorflow/core/platform:build_config_root.bzl",
11 load("//tensorflow:tensorflow.bzl", "filegroup")
14 load("//tensorflow:tensorflow.bzl", "genrule")
15 load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test")
16 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
26 "//tensorflow/compiler/xla:friends",
44 # //third_party/tensorflow/core:test_main into the same target.
50 "//tensorflow/compiler/xla:debug_options_flags",
51 "//tensorflow/core:lib",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/ |
| D | BUILD | 2 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
5 load("//tensorflow:tensorflow.bzl", "tf_py_test")
9 "//tensorflow/python/keras:__subpackages__",
10 "//tensorflow/tools/pip_package:__pkg__",
18 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
26 "//tensorflow/python:client_testlib",
27 "//tensorflow/python:dtypes",
28 "//tensorflow/python:extra_py_tests_deps",
29 "//tensorflow/python:platform_benchmark",
30 "//tensorflow/python:random_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | BUILD | 4 load("//tensorflow/compiler/tests:build_defs.bzl", "tf_xla_py_test")
7 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
10 load("//tensorflow:tensorflow.bzl", "tf_py_test")
11 load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
15 "//tensorflow/python/keras:__subpackages__",
16 "//tensorflow/tools/pip_package:__pkg__",
24 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
37 "//tensorflow/python:client_testlib",
38 "//tensorflow/python/keras",
49 "//tensorflow/python:client_testlib",
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "filegroup")
2 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
4 "//tensorflow:tensorflow.bzl",
18 "//tensorflow:internal",
27 visibility = ["//tensorflow:internal"],
30 "//tensorflow/compiler/jit:xla_kernel_creator",
40 visibility = ["//tensorflow:internal"],
47 "//tensorflow/c:c_api_internal",
48 "//tensorflow/c:tf_tensor_internal",
49 "//tensorflow/c/eager:abstract_function",
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2 load("//tensorflow:tensorflow.bzl", "if_oss", "tf_cc_test")
3 load("//tensorflow/core/profiler/builds:build_config.bzl", "tf_profiler_copts")
6 default_visibility = ["//tensorflow/core/profiler:internal"],
18 "//tensorflow/core:lib",
19 "//tensorflow/core:lib_internal",
20 "//tensorflow/core/profiler/protobuf:op_metrics_proto_cc",
21 "//tensorflow/core/profiler/protobuf:xplane_proto_cc",
22 "//tensorflow/core/profiler/utils:cost_utils",
23 "//tensorflow/core/profiler/utils:op_metrics_db_utils",
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2 load("//tensorflow:tensorflow.bzl", "if_libtpu", "tf_cc_binary", "tf_cc_test", "tf_copts", "tf_cuda…
4 "//tensorflow/core/platform/default:cuda_build_defs.bzl",
8 "//tensorflow/core/platform:build_config.bzl",
12 load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library")
15 load("//tensorflow:tensorflow.bzl", "filegroup")
17 load("//tensorflow/compiler/xla/service/cpu:build_defs.bzl", "runtime_copts")
27 "//tensorflow/compiler/aot/...",
28 "//tensorflow/compiler/jit/...",
29 "//tensorflow/compiler/mlir/...",
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "py_test")
2 load("//tensorflow/tools/test:performance.bzl", "tf_py_logged_benchmark")
5 default_visibility = ["//tensorflow:internal"],
41 "//tensorflow/python:util",
55 "//tensorflow/python:array_ops",
56 "//tensorflow/python:dtypes",
57 "//tensorflow/python:framework_ops",
58 "//tensorflow/python:math_ops",
59 "//tensorflow/python:sort_ops",
60 "//tensorflow/python:tensor_util",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 default_visibility = ["//tensorflow:internal"],
15 "//tensorflow/python:array_ops",
16 "//tensorflow/python:client_testlib",
17 "//tensorflow/python:framework_for_generated_wrappers",
18 "//tensorflow/python:framework_test_lib",
19 "//tensorflow/python:math_ops",
20 "//tensorflow/python:platform_test",
21 "//tensorflow/python/ops/linalg",
31 "//tensorflow/python:array_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | BUILD | 4 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
9 "//tensorflow/python:__pkg__",
10 "//tensorflow/python/distribute:__pkg__",
11 "//tensorflow/python/keras:__subpackages__",
12 "//tensorflow/python/training/tracking:__pkg__",
20 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
40 "//tensorflow/python:control_flow_ops",
41 "//tensorflow/python:framework",
42 "//tensorflow/python:math_ops",
43 "//tensorflow/python:resource_variable_ops",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2 load("//tensorflow:tensorflow.bzl", "tf_cc_test")
3 load("//tensorflow/core/platform:build_config.bzl", "tf_protos_all")
7 "//tensorflow/core/data:__pkg__",
8 "//tensorflow/core/data/service:__pkg__",
9 "//tensorflow/core/grappler/optimizers/data:__subpackages__",
10 "//tensorflow/core/kernels/data:__pkg__",
11 "//tensorflow/core/kernels/data/experimental:__pkg__",
50 "//tensorflow/core/kernels/data:shard_dataset_op",
51 "//tensorflow/core:framework",
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | BUILD | 6 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
7 load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_copts", "tf_cuda_library")
8 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load
9 load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") # buildifier: disable=same-origin-load
10 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
14 "//tensorflow/core/platform:build_config.bzl",
18 "//tensorflow/core/platform:build_config_root.bzl",
29 visibility = ["//tensorflow:internal"],
46 "//tensorflow/core:framework",
47 "//tensorflow/core:lib",
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
4 load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "if_mlir", "tf_cc_test")
7 load("//tensorflow:tensorflow.bzl", "if_libtpu", "if_with_tpu_support", "tf_copts")
8 load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm")
11 load("//tensorflow:tensorflow.bzl", "cc_header_only_library")
14 load("//tensorflow:tensorflow.bzl", "filegroup")
17 load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_dep…
18 load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library")
20 "//tensorflow/core/platform:build_config_root.bzl",
36 "//tensorflow/compiler/tf2xla:internal",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | BUILD | 4 load("//tensorflow:tensorflow.bzl", "tf_py_test")
5 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
9 "//tensorflow:__subpackages__",
10 "//tensorflow/python/keras:__subpackages__",
18 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
30 "//tensorflow/python:dtypes",
31 "//tensorflow/python:framework_ops",
32 "//tensorflow/python:util",
33 "//tensorflow/python:variable_scope",
34 "//tensorflow/python:variables",
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 4 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
6 "//tensorflow:tensorflow.bzl",
12 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load
13 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_tests") # buildifier: disable=same-origin-load
14 load("//tensorflow:tensorflow.bzl", "tf_grpc_dependency") # buildifier: disable=same-origin-load
15 load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") # buildifier: disable=same-origin-load
16 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
20 "//tensorflow/core/platform:build_config.bzl",
25 "//tensorflow/core/platform:build_config_root.bzl",
48 "//tensorflow/core:lib",
[all …]
|