add input argument in warpctc layer (apache#11167)

zheng-da · Jun 28, 2018 · c06e064 · c06e064
1 parent b9afbe1
commit c06e064
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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__());