| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 1 [//tensorflow/python/util:cpp_python_util] # util tfe
2 tensorflow::swig::IsNested
3 tensorflow::swig::IsNestedOrComposite
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"],
21 "//tensorflow/core:dataset_ops_op_lib",
22 "//tensorflow/core:framework",
23 "//tensorflow/core:lib",
24 "//tensorflow/core:lib_internal",
25 "//tensorflow/core/data:dataset_utils",
[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"],
22 visibility = ["//tensorflow:__subpackages__"],
34 visibility = ["//tensorflow:__subpackages__"],
42 "//tensorflow/core:experimental_dataset_ops_op_lib",
43 "//tensorflow/core:framework",
44 "//tensorflow/core/data:name_utils",
53 "//tensorflow/core:experimental_dataset_ops_op_lib",
[all …]
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "py_strict_test")
4 load("//tensorflow:tensorflow.bzl", "cuda_py_test") # buildifier: disable=same-origin-load
5 load("//tensorflow:tensorflow.bzl", "tf_py_test") # buildifier: disable=same-origin-load
8 default_visibility = ["//tensorflow:internal"],
18 "//tensorflow/python:client_testlib",
19 "//tensorflow/python:constant_op",
20 "//tensorflow/python:framework_combinations",
21 "//tensorflow/python:sparse_tensor",
22 "//tensorflow/python/data/ops:dataset_ops",
23 "//tensorflow/python/ops/ragged:ragged_tensor_value",
[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/python/distribute/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "pytype_strict_library")
2 load("//tensorflow:tensorflow.bzl", "py_test", "tf_py_test")
3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
4 load("//tensorflow/tsl/platform/default:distribute.bzl", "distribute_py_test")
5 load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
9 "//tensorflow:internal",
26 "//tensorflow/python/distribute/v1:all_reduce",
44 "//tensorflow/python:array_ops",
45 "//tensorflow/python:framework_ops",
46 "//tensorflow/python:kernels",
[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",
43 "//tensorflow/python/saved_model/registration",
[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/tsl/platform/default:distribute.bzl", "distribute_py_test")
6 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
9 default_visibility = ["//tensorflow:internal"],
14 # Used in a pybind extension whose rule must be in tensorflow/python
16 visibility = ["//tensorflow/python:__pkg__"],
20 # Used in a rule which visibility limits to tensorflow/python
22 visibility = ["//tensorflow/python:__pkg__"],
110 "//tensorflow/python:learning_rate_decay",
[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/python/kernel_tests/nn_ops/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 load("//tensorflow:tensorflow.bzl", "tf_py_test")
9 default_visibility = ["//tensorflow:internal"],
41 "//tensorflow/python:array_ops",
42 "//tensorflow/python:client_testlib",
43 "//tensorflow/python:framework_for_generated_wrappers",
44 "//tensorflow/python:nn",
45 "//tensorflow/python:nn_grad",
46 "//tensorflow/python:nn_ops",
60 "//tensorflow/python:array_ops",
[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/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__",
54 "//tensorflow/core/kernels/data:shard_dataset_op",
55 "//tensorflow/core:framework",
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | BUILD | 3 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5 "//tensorflow/core/platform:build_config.bzl",
8 load("//tensorflow:tensorflow.bzl", "if_libtpu", "tf_cc_test", "tf_copts")
9 load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependencies") # buildifier: disable=same-origin-l…
10 load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # buildifier: disable=same-origin-load
12 "//tensorflow/core/platform:build_config_root.bzl",
20 "//tensorflow/compiler/xrt/kernels:__subpackages__",
21 "//tensorflow/core/tpu:__subpackages__",
22 "//tensorflow/dtensor:__subpackages__",
60 ["//tensorflow/core/tpu/kernels:tpu_compilation_metrics"],
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | BUILD | 6 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
8 "//tensorflow:tensorflow.bzl",
14 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load
15 load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependencies") # buildifier: disable=same-origin-l…
16 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
20 "//tensorflow/core/platform:build_config.bzl",
24 "//tensorflow/core/platform:build_config_root.bzl",
30 ["//tensorflow:internal"],
38 visibility = ["//tensorflow:internal"],
55 "//tensorflow/core:framework",
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | BUILD | 1 load("//tensorflow:tensorflow.bzl", "py_strict_test")
2 load("//tensorflow:tensorflow.bzl", "py_test")
3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
4 load("//tensorflow/tools/test:performance.bzl", "tf_py_logged_benchmark")
7 default_visibility = ["//tensorflow:internal"],
46 "//tensorflow/python:util",
61 "//tensorflow/python:array_ops",
62 "//tensorflow/python:check_ops",
63 "//tensorflow/python:control_flow_ops",
64 "//tensorflow/python:data_flow_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | BUILD | 7 "//tensorflow/python/distribute:__pkg__",
8 "//tensorflow/python/feature_column:__pkg__",
9 "//tensorflow/python/keras:__subpackages__",
10 "//tensorflow/python/training/tracking:__pkg__",
11 "//tensorflow/tools/pip_package:__pkg__",
20 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
45 "//tensorflow/python/keras/engine",
46 "//tensorflow/python/keras/utils:tf_utils",
55 "//tensorflow/python:math_ops",
56 "//tensorflow/python:util",
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 2 "//tensorflow:tensorflow.bzl",
15 load("//tensorflow:tensorflow.bzl", "cc_header_only_library")
18 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
21 load("//tensorflow:tensorflow.bzl", "filegroup")
25 "//tensorflow/core/platform:build_config.bzl",
32 "//tensorflow/core/platform:rules_cc.bzl",
36 "//tensorflow/core/platform:build_config_root.bzl",
48 "//tensorflow:internal",
78 "//tensorflow/core/graph:core_cpu_headers",
79 "//tensorflow/core/public:session.h",
[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/kernel_tests/linalg/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 load("//tensorflow:tensorflow.bzl", "tf_py_test")
9 default_visibility = ["//tensorflow:internal"],
22 "//tensorflow/python:array_ops",
23 "//tensorflow/python:client_testlib",
24 "//tensorflow/python:framework_for_generated_wrappers",
25 "//tensorflow/python:linalg_ops",
26 "//tensorflow/python:math_ops",
27 "//tensorflow/python:platform",
28 "//tensorflow/python/ops/linalg",
[all …]
|
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | BUILD | 5 "//tensorflow/tools/test:performance.bzl",
10 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
13 load("//tensorflow:tensorflow.bzl", "tf_py_test")
17 "//tensorflow:internal",
50 "//tensorflow/core:protos_all_py",
51 "//tensorflow/python:array_ops",
52 "//tensorflow/python:constant_op",
53 "//tensorflow/python:dtypes",
54 "//tensorflow/python:errors",
55 "//tensorflow/python:framework_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 load("//tensorflow:tensorflow.bzl", "tf_py_test")
16 "//tensorflow/python:array_ops",
17 "//tensorflow/python:client_testlib",
18 "//tensorflow/python:cond_v2",
19 "//tensorflow/python:constant_op",
20 "//tensorflow/python:control_flow_ops",
21 "//tensorflow/python:dtypes",
22 "//tensorflow/python:framework",
23 "//tensorflow/python:framework_ops",
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | BUILD | 4 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
6 "//tensorflow:tensorflow.bzl",
13 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load
14 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_tests") # buildifier: disable=same-origin-load
15 load("//tensorflow:tensorflow.bzl", "tf_grpc_dependencies") # buildifier: disable=same-origin-load
16 load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependencies") # buildifier: disable=same-origin-l…
17 load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load
21 "//tensorflow/core/platform:build_config.bzl",
26 "//tensorflow/core/platform:build_config_root.bzl",
32 ["//tensorflow:internal"],
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 load("//tensorflow:tensorflow.bzl", "tf_py_test")
7 load("//tensorflow/core/platform:build_config_root.bzl", "tf_additional_xla_deps_py")
21 "//tensorflow/python:array_ops",
22 "//tensorflow/python:client",
23 "//tensorflow/python:client_testlib",
24 "//tensorflow/python:errors",
25 "//tensorflow/python:framework",
26 "//tensorflow/python:framework_for_generated_wrappers",
27 "//tensorflow/python:framework_test_lib",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | BUILD | 7 "//tensorflow/python:__pkg__",
8 "//tensorflow/python/feature_column:__pkg__",
9 "//tensorflow/python/keras:__subpackages__",
17 visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
47 "//tensorflow/python:py_checkpoint_reader",
48 "//tensorflow/python/data",
49 "//tensorflow/python/distribute:distribute_coordinator",
50 "//tensorflow/python/distribute:distribute_lib",
51 "//tensorflow/python/distribute:input_lib",
52 "//tensorflow/python/distribute:parameter_server_strategy",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | BUILD | 3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
6 load("//tensorflow:tensorflow.bzl", "tf_py_test")
9 default_visibility = ["//tensorflow:internal"],
24 "//tensorflow/python:client_testlib",
25 "//tensorflow/python:data_flow_ops",
26 "//tensorflow/python:errors",
27 "//tensorflow/python:framework_for_generated_wrappers",
37 "//tensorflow/python:array_ops",
38 "//tensorflow/python:client_testlib",
39 "//tensorflow/python:data_flow_ops",
[all …]
|