1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
licenses(["notice"]) # Apache 2.0
package(default_visibility = ["//tensorflow:internal"])
load("//tensorflow:tensorflow.bzl", "py_test")
py_library(
name = "tflite_lstm",
srcs = ["tflite_lstm.py"],
srcs_version = "PY2AND3",
visibility = ["//visibility:public"],
deps = [
"//tensorflow:tensorflow_py",
"//tensorflow/contrib/lite/python:lite",
"//tensorflow/python:framework",
"@six_archive//:six",
],
)
py_test(
name = "unidirectional_sequence_lstm_test",
size = "large",
srcs = ["unidirectional_sequence_lstm_test.py"],
srcs_version = "PY2AND3",
tags = [
"no_oss",
"no_pip",
],
deps = [
":tflite_lstm",
"//tensorflow:tensorflow_py",
"//tensorflow/contrib/lite/python:lite",
"//tensorflow/examples/tutorials/mnist:input_data",
"//tensorflow/python:framework_test_lib",
"//tensorflow/python:platform",
"//tensorflow/python/tools:optimize_for_inference",
"//third_party/py/numpy",
"@six_archive//:six",
],
)
|