experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Ensure that gen_rnn_ops is loaded on TF import

Browse Source 
This makes the corresponding ops available under tf.raw_ops.

PiperOrigin-RevId: 253635948
This commit is contained in:
Sergei Lebedev 2019-06-17 12:27:17 -07:00 committed by TensorFlower Gardener
parent ba04dfaf5e
commit e07ad649c5
10 changed files with 57 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/api_def/base_api/api_def_BlockLSTM.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "BlockLSTM"
visibility: HIDDEN
in_arg {
name: "seq_len_max"
description: <<END

1
tensorflow/core/api_def/base_api/api_def_BlockLSTMGrad.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "BlockLSTMGrad"
visibility: HIDDEN
in_arg {
name: "seq_len_max"
description: <<END

1
tensorflow/core/api_def/base_api/api_def_GRUBlockCell.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "GRUBlockCell"
visibility: HIDDEN
summary: "Computes the GRU cell forward propagation for 1 time step."
description: <<END
Args

1
tensorflow/core/api_def/base_api/api_def_GRUBlockCellGrad.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "GRUBlockCellGrad"
visibility: HIDDEN
summary: "Computes the GRU cell back-propagation for 1 time step."
description: <<END
Args

1
tensorflow/core/api_def/base_api/api_def_LSTMBlockCell.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "LSTMBlockCell"
visibility: HIDDEN
in_arg {
name: "x"
description: <<END

1
tensorflow/core/api_def/base_api/api_def_LSTMBlockCellGrad.pbtxt
View File

@ -1,5 +1,6 @@
op {
graph_op_name: "LSTMBlockCellGrad"
visibility: HIDDEN
in_arg {
name: "x"
description: <<END

1
tensorflow/python/BUILD
View File

@ -154,6 +154,7 @@ py_library(
":platform",
":proto_ops",
":pywrap_tensorflow",
":rnn_ops_gen",
":saver_test_utils",
":script_ops",
":session_ops",

2
tensorflow/python/__init__.py
View File

@ -115,6 +115,8 @@ from tensorflow.python.ops import gen_boosted_trees_ops as _gen_boosted_trees_op
# Import cudnn rnn ops to make sure their ops are registered.
from tensorflow.python.ops import gen_cudnn_rnn_ops as _
# Import rnn_ops to make sure their ops are registered.
from tensorflow.python.ops import gen_rnn_ops as _
# Import the names from python/training.py as train.Name.
from tensorflow.python.training import training as train

24
tensorflow/tools/api/golden/v1/tensorflow.raw_ops.pbtxt
View File

@ -440,6 +440,14 @@ tf_module {
name: "BitwiseXor"
argspec: "args=[\'x\', \'y\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "BlockLSTM"
argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'3\', \'False\', \'None\'], "
}
member_method {
name: "BlockLSTMGrad"
argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'i\', \'cs\', \'f\', \'o\', \'ci\', \'co\', \'h\', \'cs_grad\', \'h_grad\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "BoostedTreesAggregateStats"
argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'feature\', \'max_splits\', \'num_buckets\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
@ -1444,6 +1452,14 @@ tf_module {
name: "FusedResizeAndPadConv2D"
argspec: "args=[\'input\', \'size\', \'paddings\', \'filter\', \'mode\', \'strides\', \'padding\', \'resize_align_corners\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
}
member_method {
name: "GRUBlockCell"
argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "GRUBlockCellGrad"
argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'r\', \'u\', \'c\', \'d_h\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "Gather"
argspec: "args=[\'params\', \'indices\', \'validate_indices\', \'name\'], varargs=None, keywords=None, defaults=[\'True\', \'None\'], "
@ -1728,6 +1744,14 @@ tf_module {
name: "LRNGrad"
argspec: "args=[\'input_grads\', \'input_image\', \'output_image\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=None, keywords=None, defaults=[\'5\', \'1\', \'1\', \'0.5\', \'None\'], "
}
member_method {
name: "LSTMBlockCell"
argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'3\', \'False\', \'None\'], "
}
member_method {
name: "LSTMBlockCellGrad"
argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'i\', \'cs\', \'f\', \'o\', \'ci\', \'co\', \'cs_grad\', \'h_grad\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "LeakyRelu"
argspec: "args=[\'features\', \'alpha\', \'name\'], varargs=None, keywords=None, defaults=[\'0.2\', \'None\'], "

24
tensorflow/tools/api/golden/v2/tensorflow.raw_ops.pbtxt
View File

@ -440,6 +440,14 @@ tf_module {
name: "BitwiseXor"
argspec: "args=[\'x\', \'y\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "BlockLSTM"
argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'3\', \'False\', \'None\'], "
}
member_method {
name: "BlockLSTMGrad"
argspec: "args=[\'seq_len_max\', \'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'i\', \'cs\', \'f\', \'o\', \'ci\', \'co\', \'h\', \'cs_grad\', \'h_grad\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "BoostedTreesAggregateStats"
argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'feature\', \'max_splits\', \'num_buckets\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
@ -1444,6 +1452,14 @@ tf_module {
name: "FusedResizeAndPadConv2D"
argspec: "args=[\'input\', \'size\', \'paddings\', \'filter\', \'mode\', \'strides\', \'padding\', \'resize_align_corners\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
}
member_method {
name: "GRUBlockCell"
argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "GRUBlockCellGrad"
argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'r\', \'u\', \'c\', \'d_h\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "Gather"
argspec: "args=[\'params\', \'indices\', \'validate_indices\', \'name\'], varargs=None, keywords=None, defaults=[\'True\', \'None\'], "
@ -1728,6 +1744,14 @@ tf_module {
name: "LRNGrad"
argspec: "args=[\'input_grads\', \'input_image\', \'output_image\', \'depth_radius\', \'bias\', \'alpha\', \'beta\', \'name\'], varargs=None, keywords=None, defaults=[\'5\', \'1\', \'1\', \'0.5\', \'None\'], "
}
member_method {
name: "LSTMBlockCell"
argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'forget_bias\', \'cell_clip\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'1\', \'3\', \'False\', \'None\'], "
}
member_method {
name: "LSTMBlockCellGrad"
argspec: "args=[\'x\', \'cs_prev\', \'h_prev\', \'w\', \'wci\', \'wcf\', \'wco\', \'b\', \'i\', \'cs\', \'f\', \'o\', \'ci\', \'co\', \'cs_grad\', \'h_grad\', \'use_peephole\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "LeakyRelu"
argspec: "args=[\'features\', \'alpha\', \'name\'], varargs=None, keywords=None, defaults=[\'0.2\', \'None\'], "