| /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/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/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/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/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/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/saved_model/ |
| D | BUILD | 2 # TensorFlow SavedModel.
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",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 6 "//tensorflow/core/platform:build_config.bzl",
10 "//tensorflow/core/platform:build_config_root.bzl",
14 "//tensorflow:tensorflow.bzl",
26 "//tensorflow/stream_executor:build_defs.bzl",
30 "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
35 load("//tensorflow:tensorflow.bzl", "filegroup")
38 load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud")
41 load("//tensorflow:tensorflow.bzl", "if_nccl")
51 "//tensorflow/compiler/xla:friends",
96 "//tensorflow/compiler/xla:xla_data_proto",
[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/service/gpu/tests/ |
| D | BUILD | 7 load("//tensorflow:tensorflow.bzl", "filegroup")
8 load("//tensorflow/compiler/xla/tests:build_defs.bzl", "xla_test")
9 load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test")
11 "//tensorflow/core/platform:build_config_root.bzl",
14 load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
16 "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
32 "//tensorflow/compiler/xla:friends",
52 "//tensorflow/compiler/xla:debug_options_flags",
53 "//tensorflow/compiler/xla:shape_util",
54 "//tensorflow/compiler/xla:types",
[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/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/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/distributed_runtime/ |
| D | BUILD | 2 # A distributed runtime for TensorFlow, which allows graph execution
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"],
[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", "tf_profiler_pybind…
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/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/compiler/tf2xla/ |
| D | BUILD | 1 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
2 load("//tensorflow:tensorflow.bzl", "if_google", "if_libtpu", "tf_cc_binary", "tf_cc_test", "tf_cop…
3 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
10 "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
14 "//tensorflow/core/platform:build_config.bzl",
18 load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library")
21 load("//tensorflow:tensorflow.bzl", "filegroup")
24 load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable")
26 load("//tensorflow/compiler/xla/service/cpu:build_defs.bzl", "runtime_copts")
36 "//tensorflow/compiler/aot/...",
[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", "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/kernel_tests/nn_ops/ |
| D | BUILD | 1 # Tests of TensorFlow NN kernels written using the Python API.
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",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | BUILD | 1 # Tests of TensorFlow linalg kernels written using the Python API.
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",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | BUILD | 2 # Contains the Keras layers (internal TensorFlow version).
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",
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | BUILD | 4 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
5 load("//tensorflow/compiler/xla/tests:build_defs.bzl", "xla_test")
6 load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library", "xla_py_test_deps")
8 "//tensorflow/core/platform:build_config.bzl",
11 load("//tensorflow:tensorflow.bzl", "if_libtpu", "tf_cc_test")
14 load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable")
17 load("//tensorflow:tensorflow.bzl", "filegroup")
20 load("//tensorflow:tensorflow.bzl", "internal_hlo_deps")
23 load("//tensorflow:tensorflow.bzl", "internal_cuda_deps")
25 "//tensorflow/tsl/platform/default:cuda_build_defs.bzl",
[all …]
|