BUILD 1.27 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
licenses(["notice"])

package(
    default_visibility = ["//:friends"],  # sequence projection
)

py_library(
    name = "prado",
    srcs = ["prado.py"],
    srcs_version = "PY3",
    deps = [
        # package absl/logging
        # package tensorflow
        "//layers:base_layers",  # sequence projection
        "//layers:conv_layers",  # sequence projection
        "//layers:dense_layers",  # sequence projection
        "//layers:projection_layers",  # sequence projection
        "//layers:quantization_layers",  # sequence projection
        # "//tf_ops:tf_custom_ops"  # sequence projection
        "//tf_ops:tf_custom_ops_py",  # sequence projection
    ],
)
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

py_library(
    name = "pqrnn",
    srcs = ["pqrnn.py"],
    srcs_version = "PY3",
    deps = [
        # package absl/logging
        # package tensorflow
        "//layers:base_layers",  # sequence projection
        "//layers:dense_layers",  # sequence projection
        "//layers:misc_layers",  # sequence projection
        "//layers:projection_layers",  # sequence projection
        "//layers:qrnn_layers",  # sequence projection
        "//layers:quantization_layers",  # sequence projection
        # "//tf_ops:tf_custom_ops"  # sequence projection
        "//tf_ops:tf_custom_ops_py",  # sequence projection
    ],
)