55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
# Description:
|
|
# Contains the legacy TF RNN APIs (internal TensorFlow version).
|
|
|
|
package(
|
|
default_visibility = ["//tensorflow:__subpackages__"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
py_library(
|
|
name = "rnn_cell_impl",
|
|
srcs = ["rnn_cell_impl.py"],
|
|
deps = [
|
|
":rnn_cell_wrapper_impl",
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:clip_ops",
|
|
"//tensorflow/python:constant_op",
|
|
"//tensorflow/python:dtypes",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:init_ops",
|
|
"//tensorflow/python:layers_base",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:partitioned_variables",
|
|
"//tensorflow/python:platform",
|
|
"//tensorflow/python:tensor_shape",
|
|
"//tensorflow/python:tensor_util",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python:variable_scope",
|
|
"//tensorflow/python:variables",
|
|
"//tensorflow/python/eager:context",
|
|
"//tensorflow/python/keras:activations",
|
|
"//tensorflow/python/keras:initializers",
|
|
"//tensorflow/python/keras/engine:input_spec",
|
|
"//tensorflow/python/keras/utils:tf_utils",
|
|
"//tensorflow/python/training/tracking:base",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name = "rnn_cell_wrapper_impl",
|
|
srcs = ["rnn_cell_wrapper_impl.py"],
|
|
deps = [
|
|
"//tensorflow/python:array_ops",
|
|
"//tensorflow/python:framework_ops",
|
|
"//tensorflow/python:math_ops",
|
|
"//tensorflow/python:nn_ops",
|
|
"//tensorflow/python:random_ops",
|
|
"//tensorflow/python:tensor_array_ops",
|
|
"//tensorflow/python:tensor_shape",
|
|
"//tensorflow/python:tensor_util",
|
|
"//tensorflow/python:util",
|
|
"//tensorflow/python/keras/utils:generic_utils",
|
|
],
|
|
)
|