[GitHub] TaoLv commented on issue #10410: Fix output names of nn operators.

[GitBox <gi...@apache.org>]

TaoLv commented on issue #10410: Fix output names of nn operators.
URL: https://github.com/apache/incubator-mxnet/pull/10410#issuecomment-378977002
 
 
   Please also help fix that in pooling.cc. When requires workspace, there will be two outputs:
   ```
   .set_attr<nnvm::FListOutputNames>("FListOutputNames",
       [](const NodeAttrs& attrs) {
     const PoolingParam &param = nnvm::get<PoolingParam>(attrs.parsed);
     if (GetNumOutputs(param) == 2)
       return std::vector<std::string>{"output", "workspace"};
     else
       return std::vector<std::string>{"output"};
   })
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services