Home
last modified time | relevance | path

Searched refs:tensorflow (Results 1 – 25 of 7188) sorted by relevance

12345678910>>...288

/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt2 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::IsMappingView
9 tensorflow::swig::IsAttrs
10 tensorflow::swig::IsTensor
11 tensorflow::swig::IsResourceVariable
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
DBUILD3 load("//tensorflow:tensorflow.bzl", "sycl_py_test", "tf_custom_op_library", "tf_py_test")
4 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
7 default_visibility = ["//tensorflow:internal"],
21 "//tensorflow/python:array_ops",
22 "//tensorflow/python:client_testlib",
23 "//tensorflow/python:framework_for_generated_wrappers",
24 "//tensorflow/python:string_ops",
34 "//tensorflow/python:array_ops",
35 "//tensorflow/python:client_testlib",
36 "//tensorflow/python:framework_for_generated_wrappers",
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
DBUILD3 load("//tensorflow:tensorflow.bzl", "tf_py_test")
4 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
7 default_visibility = ["//tensorflow:internal"],
17 "//tensorflow/python:array_ops",
18 "//tensorflow/python:client_testlib",
19 "//tensorflow/python:dtypes",
20 "//tensorflow/python:errors",
21 "//tensorflow/python:math_ops",
22 "//tensorflow/python:sparse_tensor",
23 "//tensorflow/python/data/ops:dataset_ops",
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
DBUILD1 load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cc_test_mkl", "tf_kernel_library")
2 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
6 "//tensorflow/core/platform:build_config_root.bzl",
21 "//tensorflow/python:__pkg__",
33 "//tensorflow/core:lib",
34 "//tensorflow/core:protos_all_cc",
35 "//tensorflow/core/grappler:grappler_item",
36 "//tensorflow/core/grappler:op_types",
37 "//tensorflow/core/grappler:utils",
38 "//tensorflow/core/grappler/clusters:cluster",
[all …]
/external/tensorflow/tensorflow/python/distribute/
DBUILD1 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")
6 default_visibility = ["//tensorflow:internal"],
22 "//tensorflow/python/keras/distribute:keras_correctness_test_lib",
23 "//tensorflow/python/keras/distribute:keras_test_lib",
34 "//tensorflow/python:array_ops",
35 "//tensorflow/python:framework_ops",
36 "//tensorflow/python:math_ops",
37 "//tensorflow/python:nccl_ops",
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
DBUILD1 load("//tensorflow:tensorflow.bzl", "tf_py_test")
2 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
5 default_visibility = ["//tensorflow:internal"],
21 "//tensorflow/python:client_testlib",
22 "//tensorflow/python:errors",
23 "//tensorflow/python/data/experimental/ops:testing",
24 "//tensorflow/python/data/kernel_tests:test_base",
25 "//tensorflow/python/data/ops:dataset_ops",
38 "//tensorflow/python:client_testlib",
39 "//tensorflow/python/data/experimental/ops:distribute",
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
DBUILD5 "//tensorflow:tensorflow.bzl",
20 deps = ["//tensorflow/core:framework"],
38 "//tensorflow/core:core_cpu",
39 "//tensorflow/core:core_cpu_internal",
40 "//tensorflow/core:framework",
41 "//tensorflow/core:framework_internal",
42 "//tensorflow/core:lib",
43 "//tensorflow/core:lib_internal",
44 "//tensorflow/core:protos_all_cc",
45 "//tensorflow/core:tensor_testutil",
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
DBUILD5 "//tensorflow:tensorflow.bzl",
11 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
12 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_tests")
16 "//tensorflow/core/platform:build_config.bzl",
20 "//tensorflow/core/platform:build_config_root.bzl",
45 "//tensorflow:grpc",
46 "//tensorflow:grpc++",
47 "//tensorflow/core:lib",
49 "//tensorflow/core/distributed_runtime:tensor_coding",
50 "//tensorflow/core:lib_internal",
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
DBUILD1 load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test", "tf_cuda_cc_test")
3 "//tensorflow/core/platform/default:cuda_build_defs.bzl",
7 "//tensorflow/core/platform:build_config.bzl",
11 load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library")
12 load("//tensorflow:tensorflow.bzl", "tf_portable_proto_library")
23 "//tensorflow/compiler/aot/...",
24 "//tensorflow/compiler/jit/...",
25 "//tensorflow/compiler/mlir/...",
26 "//tensorflow/compiler/tests/...",
27 "//tensorflow/compiler/tf2xla/...",
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
DBUILD1 load("//tensorflow:tensorflow.bzl", "tf_cc_test")
2 load("//tensorflow/core/platform:build_config.bzl", "tf_protos_all")
6 "//tensorflow/core/grappler/optimizers/data:__subpackages__",
7 "//tensorflow/core/kernels/data:__pkg__",
8 "//tensorflow/core/kernels/data/experimental:__pkg__",
46 "//tensorflow/core/grappler:grappler_item",
47 "//tensorflow/core/grappler:mutable_graph_view",
48 "//tensorflow/core/grappler:op_types",
49 "//tensorflow/core/grappler/clusters:cluster",
50 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
DBUILD5 "//tensorflow:tensorflow.bzl",
11 default_visibility = ["//tensorflow:internal"],
22 "//tensorflow/core:experimental_dataset_ops_op_lib",
23 "//tensorflow/core:framework",
24 "//tensorflow/core/kernels/data:name_utils",
35 "//tensorflow/core:experimental_dataset_ops_op_lib",
36 "//tensorflow/core:test_main",
37 "//tensorflow/core/kernels/data:dataset_test_base",
38 "//tensorflow/core/kernels/data:range_dataset_op",
39 "//tensorflow/core/kernels/data:take_dataset_op",
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
DBUILD4 load("//tensorflow/compiler/xla/tests:build_defs.bzl", "generate_backend_suites", "generate_backend…
6 "//tensorflow/core/platform:build_config_root.bzl",
9 load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test")
19 "//tensorflow/compiler/xla:friends",
37 # //third_party/tensorflow/core:test_main into the same target.
43 "//tensorflow/compiler/xla:debug_options_flags",
44 "//tensorflow/core:lib",
45 "//tensorflow/core:test",
56 "//tensorflow/compiler/xla:types",
57 "//tensorflow/core:test",
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
DBUILD3 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/core/distributed_runtime/
DBUILD6 load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_copts", "tf_cuda_library")
7 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
11 "//tensorflow/core/platform:build_config.bzl",
15 "//tensorflow/core/platform:build_config_root.bzl",
28 visibility = ["//tensorflow:internal"],
45 "//tensorflow/core:framework",
46 "//tensorflow/core:lib",
56 "//tensorflow/core:lib",
57 "//tensorflow/core:test",
58 "//tensorflow/core:test_main",
[all …]
/external/tensorflow/tensorflow/compiler/jit/
DBUILD1 load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "if_mlir", "tf_cc_test")
2 load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm")
3 load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_dep…
4 load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library")
5 load("//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags")
15 "//tensorflow/compiler/tf2xla:internal",
22 "//tensorflow/compiler/tf2xla:friends",
47 "//tensorflow/compiler/plugin",
61 "//tensorflow/compiler/jit/kernels:xla_ops",
62 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD5 "//tensorflow/core/platform:build_config.bzl",
9 "//tensorflow/core/platform:build_config_root.bzl",
13 "//tensorflow:tensorflow.bzl",
20 "//tensorflow/core/platform/default:cuda_build_defs.bzl",
27 load("//tensorflow:tensorflow.bzl", "if_nccl")
37 "//tensorflow/compiler/xla:friends",
54 protodeps = ["//tensorflow/compiler/xla:xla_data_proto"],
62 "//tensorflow/compiler/xla:types",
63 "//tensorflow/core:framework",
84 "//tensorflow/compiler/xla:shape_util",
[all …]
/external/tensorflow/tensorflow/compiler/tests/
DBUILD1 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
2 load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test")
3 load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library")
4 load("//tensorflow/compiler/tests:build_defs.bzl", "tf_xla_py_test")
5 load("//tensorflow/compiler/tests:build_defs.bzl", "generate_backend_suites")
7 "//tensorflow/core/platform:build_config_root.bzl",
19 "//tensorflow/compiler/tf2xla:internal",
26 "//tensorflow/compiler/tf2xla:friends",
39 "//tensorflow/core:protos_all_py",
40 "//tensorflow/python:array_ops",
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
DBUILD2 "//tensorflow:tensorflow.bzl",
15 "//tensorflow:internal",
29 visibility = ["//tensorflow:internal"],
31 "//tensorflow:android": [
32 "//tensorflow/core:android_tensorflow_lib_lite",
35 "//tensorflow/core:core_cpu_lib",
36 "//tensorflow/core:framework",
37 "//tensorflow/core:framework_internal",
38 "//tensorflow/core:lib",
39 "//tensorflow/core:lib_internal",
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
DBUILD4 load("//tensorflow:tensorflow.bzl", "tf_py_test")
5 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
42 "//tensorflow/python/feature_column:feature_column_py",
43 "//tensorflow/python/keras/layers/preprocessing",
44 "//tensorflow/python/keras/utils:tf_utils",
53 "//tensorflow/python:math_ops",
54 "//tensorflow/python:util",
55 "//tensorflow/python/keras:backend",
56 "//tensorflow/python/keras:base_layer",
57 "//tensorflow/python/keras:constraints",
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
DBUILD2 default_visibility = ["//tensorflow:internal"],
13 "//tensorflow/python:dtypes",
14 "//tensorflow/python:experimental_dataset_ops_gen",
15 "//tensorflow/python:framework_ops",
16 "//tensorflow/python:tensor_shape",
17 "//tensorflow/python:tensor_util",
18 "//tensorflow/python:util",
19 "//tensorflow/python/data/ops:dataset_ops",
20 "//tensorflow/python/data/util:convert",
21 "//tensorflow/python/data/util:nest",
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
DBUILD3 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/distributions",
34 "//tensorflow/python:array_ops",
[all …]
/external/tensorflow/tensorflow/python/saved_model/
DBUILD4 load("//tensorflow:tensorflow.bzl", "cuda_py_test")
5 load("//tensorflow:tensorflow.bzl", "tf_py_test")
8 # TODO(drpng): change that to //third_party/tensorflow:internal
20 visibility = ["//tensorflow:internal"],
34 "//tensorflow/python:util",
42 deps = ["//tensorflow/python:util"],
49 deps = ["//tensorflow/python:util"],
56 deps = ["//tensorflow/python:util"],
69 "//tensorflow/core:protos_all_py",
70 "//tensorflow/python:framework_for_generated_wrappers",
[all …]
/external/tensorflow/tensorflow/
Dopensource_only.files1 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/core/public/version.h
8 tensorflow/lite/micro/build_def.bzl
9 tensorflow/python/autograph/core/config.py
10 tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
DBUILD1 load("//tensorflow:tensorflow.bzl", "py_test")
2 load("//tensorflow/tools/test:performance.bzl", "tf_py_logged_benchmark")
13 "//tensorflow:internal",
51 "//tensorflow/python:util",
65 "//tensorflow/python:array_ops",
66 "//tensorflow/python:dtypes",
67 "//tensorflow/python:framework_ops",
68 "//tensorflow/python:math_ops",
69 "//tensorflow/python:sort_ops",
70 "//tensorflow/python:tensor_util",
[all …]
/external/tensorflow/tensorflow/python/feature_column/
DBUILD1 load("//tensorflow:tensorflow.bzl", "py_test", "tf_py_test")
4 default_visibility = ["//tensorflow:internal"],
15 "//tensorflow/python:util",
25 "//tensorflow/python:array_ops",
26 "//tensorflow/python:check_ops",
27 "//tensorflow/python:control_flow_ops",
28 "//tensorflow/python:dtypes",
29 "//tensorflow/python:embedding_ops",
30 "//tensorflow/python:framework_ops",
31 "//tensorflow/python:init_ops",
[all …]

12345678910>>...288