This is an automated email from the ASF dual-hosted git repository.
jxie pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new aafbd17 add input argument in warpctc layer (#11167)
aafbd17 is described below
commit aafbd17460da31a31414a501f99a3e4a2346efe9
Author: Wang Jiajun <[email protected]>
AuthorDate: Thu Jun 7 01:57:50 2018 +0800
add input argument in warpctc layer (#11167)
---
plugin/warpctc/warpctc.cc | 2 ++
1 file changed, 2 insertions(+)
diff --git a/plugin/warpctc/warpctc.cc b/plugin/warpctc/warpctc.cc
index 055a6d6..aac36a3 100644
--- a/plugin/warpctc/warpctc.cc
+++ b/plugin/warpctc/warpctc.cc
@@ -41,6 +41,8 @@ Operator *WarpCTCProp::CreateOperator(Context ctx) const {
DMLC_REGISTER_PARAMETER(WarpCTCParam);
MXNET_REGISTER_OP_PROPERTY(WarpCTC, WarpCTCProp)
+.add_argument("data", "NDArray-or-Symbol", "Input data.")
+.add_argument("label", "NDArray-or-Symbol", "Input label.")
.describe("warp ctc.")
.add_arguments(WarpCTCParam::__FIELDS__());
--
To stop receiving notification emails like this one, please contact
[email protected].
