Skip to content

Commit c705568

Browse files
bignamehypnealwu
authored andcommitted
Add different rnn implementation modes to ptb tutorial (tensorflow#2276)
1 parent 7e9e15a commit c705568

File tree

4 files changed

+302
-67
lines changed

4 files changed

+302
-67
lines changed

tutorials/rnn/ptb/BUILD

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,13 @@ py_test(
3636
],
3737
)
3838

39+
py_library(
40+
name = "util",
41+
srcs = ["util.py"],
42+
srcs_version = "PY2AND3",
43+
deps = ["//tensorflow:tensorflow_py"],
44+
)
45+
3946
py_binary(
4047
name = "ptb_word_lm",
4148
srcs = [
@@ -44,7 +51,8 @@ py_binary(
4451
srcs_version = "PY2AND3",
4552
deps = [
4653
":reader",
47-
"//tensorflow:tensorflow_py",
54+
":util",
55+
"//tensorflow:tensorflow_py,
4856
],
4957
)
5058

tutorials/rnn/ptb/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,3 +19,4 @@
1919
from __future__ import print_function
2020

2121
import reader
22+
import util

0 commit comments

Comments
 (0)