| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | BUILD | 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",
37 "//tensorflow/python:array_ops",
38 "//tensorflow/python:client_testlib",
39 "//tensorflow/python:dtypes",
40 "//tensorflow/python:errors",
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | BUILD | 5 load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
26 "//tensorflow/python/distribute/v1:all_reduce",
44 "//tensorflow/python:array_ops",
45 "//tensorflow/python:framework_ops",
46 "//tensorflow/python:kernels",
47 "//tensorflow/python:math_ops",
48 "//tensorflow/python:platform",
49 "//tensorflow/python:resource_variable_ops",
50 "//tensorflow/python:tensor_util",
51 "//tensorflow/python/client:device_lib",
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | BUILD | 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",
111 "//tensorflow/python:sdca_ops",
112 "//tensorflow/python/checkpoint:checkpoint_management",
113 "//tensorflow/python/trackable:base_delegate",
114 "//tensorflow/python/training/experimental:loss_scale_optimizer",
115 "//tensorflow/python/training/experimental:mixed_precision",
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | BUILD | 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",
31 "//tensorflow/python/data/kernel_tests:test_base",
32 "//tensorflow/python/data/ops:dataset_ops",
41 "//tensorflow/python:client_testlib",
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | BUILD | 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",
65 "//tensorflow/python:dtypes",
66 "//tensorflow/python:framework_ops",
67 "//tensorflow/python:math_ops",
68 "//tensorflow/python:ragged_array_ops_gen",
69 "//tensorflow/python:sort_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | BUILD | 1 # Tests of TensorFlow linalg kernels written using the Python API.
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",
38 "//tensorflow/python:client_testlib",
39 "//tensorflow/python:framework_for_generated_wrappers",
[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__",
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",
57 "//tensorflow/python/keras:backend",
[all …]
|
| D | __init__.py | 17 from tensorflow.python import tf2
22 from tensorflow.python.keras.engine.input_layer import Input
23 from tensorflow.python.keras.engine.input_layer import InputLayer
24 from tensorflow.python.keras.engine.input_spec import InputSpec
25 from tensorflow.python.keras.engine.base_layer import Layer
26 from tensorflow.python.keras.engine.base_preprocessing_layer import PreprocessingLayer
29 from tensorflow.python.keras.layers.advanced_activations import LeakyReLU
30 from tensorflow.python.keras.layers.advanced_activations import PReLU
31 from tensorflow.python.keras.layers.advanced_activations import ELU
32 from tensorflow.python.keras.layers.advanced_activations import ReLU
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | BUILD | 1 # Tests of TensorFlow NN kernels written using the Python API.
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",
61 "//tensorflow/python:client_testlib",
62 "//tensorflow/python:framework_for_generated_wrappers",
[all …]
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | BUILD | 41 "//tensorflow/python:util",
42 "//tensorflow/python/saved_model/model_utils",
43 "//tensorflow/python/saved_model/registration",
53 "//tensorflow/python/util:tf_export",
61 deps = ["//tensorflow/python/util:tf_export"],
68 deps = ["//tensorflow/python/util:tf_export"],
84 "//tensorflow/python:framework_for_generated_wrappers",
85 "//tensorflow/python:lib",
86 "//tensorflow/python:platform",
87 "//tensorflow/python:saver",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | BUILD | 1 # Tests of TensorFlow array ops kernels written using the Python API.
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",
28 "//tensorflow/python:gradients",
29 "//tensorflow/python:map_fn",
[all …]
|
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | BUILD | 51 "//tensorflow/python:array_ops",
52 "//tensorflow/python:constant_op",
53 "//tensorflow/python:dtypes",
54 "//tensorflow/python:errors",
55 "//tensorflow/python:framework_ops",
56 "//tensorflow/python:init_ops",
57 "//tensorflow/python:io_ops_gen",
58 "//tensorflow/python:lib",
59 "//tensorflow/python:platform",
60 "//tensorflow/python:pywrap_tensorflow",
[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/tpu/ |
| D | BUILD | 6 load("//tensorflow/python/tpu:tpu.bzl", "tpu_py_test")
39 "//tensorflow/python:client_testlib",
40 "//tensorflow/python:platform",
47 actual = "//tensorflow/python/tpu/ops",
55 "//tensorflow/python:array_ops",
56 "//tensorflow/python:control_flow_ops",
57 "//tensorflow/python:framework_for_generated_wrappers",
58 "//tensorflow/python:init_ops",
59 "//tensorflow/python:math_ops",
60 "//tensorflow/python:platform",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | BUILD | 1 # Tests of TensorFlow control flow ops written using the Python API.
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",
24 "//tensorflow/python:gradients",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | BUILD | 1 # Tests of TensorFlow kernels written using the Python API.
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",
33 "//tensorflow/python:array_ops",
34 "//tensorflow/python:client_testlib",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | BUILD | 1 # Tests of TensorFlow kernels written using the Python API.
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",
40 "//tensorflow/python:errors",
41 "//tensorflow/python:framework_for_generated_wrappers",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | BUILD | 1 # Tests of TensorFlow math kernels written using the Python API.
21 "//tensorflow/python:array_ops",
22 "//tensorflow/python:client_testlib",
23 "//tensorflow/python:framework_for_generated_wrappers",
24 "//tensorflow/python:math_ops",
25 "//tensorflow/python:string_ops",
36 "//tensorflow/python:client_testlib",
37 "//tensorflow/python:nn_ops",
53 "//tensorflow/python:client_testlib",
54 "//tensorflow/python:math_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | BUILD | 11 "//tensorflow/python:dtypes",
12 "//tensorflow/python:experimental_dataset_ops_gen",
13 "//tensorflow/python:framework_ops",
14 "//tensorflow/python:tensor_shape",
15 "//tensorflow/python:tensor_util",
16 "//tensorflow/python:util",
17 "//tensorflow/python/data/ops:dataset_ops",
18 "//tensorflow/python/data/util:convert",
19 "//tensorflow/python/data/util:nest",
20 "//tensorflow/python/data/util:structure",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | BUILD | 1 # Tests of TensorFlow sparse ops written using the Python API.
18 "//tensorflow/python:client",
19 "//tensorflow/python:client_testlib",
20 "//tensorflow/python:framework",
21 "//tensorflow/python:framework_for_generated_wrappers",
22 "//tensorflow/python:math_ops",
23 "//tensorflow/python:sparse_grad",
24 "//tensorflow/python:sparse_ops",
34 "//tensorflow/python:array_ops",
35 "//tensorflow/python:client_testlib",
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | BUILD | 15 "//tensorflow/python:client_testlib",
16 "//tensorflow/python:constant_op",
17 "//tensorflow/python:dtypes",
18 "//tensorflow/python:errors",
19 "//tensorflow/python:math_ops",
20 "//tensorflow/python/data/experimental/ops:testing",
21 "//tensorflow/python/data/kernel_tests:test_base",
22 "//tensorflow/python/data/ops:dataset_ops",
23 "//tensorflow/python/data/ops:options",
34 "//tensorflow/python:array_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | BUILD | 1 # Tests of TensorFlow kernels written using the Python API.
23 "//tensorflow/python:client",
24 "//tensorflow/python:client_testlib",
25 "//tensorflow/python:framework_for_generated_wrappers",
26 "//tensorflow/python:platform",
27 "//tensorflow/python:platform_benchmark",
39 "//tensorflow/python:array_ops",
40 "//tensorflow/python:check_ops",
41 "//tensorflow/python:client_testlib",
42 "//tensorflow/python:framework",
[all …]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | BUILD | 13 "//tensorflow/python/keras/google/private_tf_api_test:__pkg__",
34 "//tensorflow/python:training",
35 "//tensorflow/python/keras/distribute",
36 "//tensorflow/python/keras/layers",
37 "//tensorflow/python/keras/mixed_precision:mixed_precision_experimental",
38 "//tensorflow/python/keras/optimizer_v2",
39 "//tensorflow/python/keras/saving",
40 "//tensorflow/python/keras/utils",
41 "//tensorflow/python/saved_model",
52 "//tensorflow/python:array_ops",
[all …]
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | BUILD | 11 "//tensorflow/python:framework_ops",
12 "//tensorflow/python:function",
13 "//tensorflow/python:util",
14 "//tensorflow/python/data/util:nest",
15 "//tensorflow/python/data/util:structure",
29 "//tensorflow/python:constant_op",
30 "//tensorflow/python:control_flow_ops",
31 "//tensorflow/python:dataset_ops_gen",
32 "//tensorflow/python:dtypes",
33 "//tensorflow/python:experimental_dataset_ops_gen",
[all …]
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/tests/ |
| D | BUILD | 35 "//tensorflow/python:array_ops",
36 "//tensorflow/python:dtypes",
37 "//tensorflow/python:math_ops",
38 "//tensorflow/python:session",
39 "//tensorflow/python:variable_scope",
40 "//tensorflow/python:variables",
41 "//tensorflow/python/compat:v2_compat",
42 "//tensorflow/python/distribute:input_lib",
43 "//tensorflow/python/saved_model:builder",
44 "//tensorflow/python/saved_model:save",
[all …]
|