提交 aab7cead authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Align with the commited names.

上级 94199f17
...@@ -2173,8 +2173,8 @@ def _split_rnn_params(w, desc, layer, input_size, dtype, rnn_mode): ...@@ -2173,8 +2173,8 @@ def _split_rnn_params(w, desc, layer, input_size, dtype, rnn_mode):
class GpuDnnRNNOp(DnnBase): class GpuDnnRNNOp(DnnBase):
__props__ = () __props__ = ()
_cop_numi = 5 _cop_num_inputs = 5
_cop_numo = 4 _cop_num_outputs = 4
def __init__(self, rnn_mode, direction_mode): def __init__(self, rnn_mode, direction_mode):
DnnBase.__init__(self, ["dnn_rnn_fwd.c"], 'dnn_rnn_fwd') DnnBase.__init__(self, ["dnn_rnn_fwd.c"], 'dnn_rnn_fwd')
...@@ -2245,8 +2245,8 @@ class GpuDnnRNNOp(DnnBase): ...@@ -2245,8 +2245,8 @@ class GpuDnnRNNOp(DnnBase):
class GpuDnnRNNGradInputs(DnnBase): class GpuDnnRNNGradInputs(DnnBase):
__props__ = () __props__ = ()
_cop_numi = 10 _cop_num_inputs = 10
_cop_numo = 4 _cop_num_outputs = 4
def __init__(self): def __init__(self):
DnnBase.__init__(self, ['dnn_rnn_gi.c'], 'dnn_rnn_gi') DnnBase.__init__(self, ['dnn_rnn_gi.c'], 'dnn_rnn_gi')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论