-
Notifications
You must be signed in to change notification settings - Fork 6.8k
enable setting return type for operators in registry #1390
Conversation
Can this be accomplished by exposing num visible outputs? |
@piiswrong nope. The number of outputs for slice_channel is not fixed. It depends on the parameters. |
Is the return type of Slice still an array if there is only one output? |
virtual void Forward(const OpContext &ctx,
const std::vector<TBlob> &in_data,
const std::vector<OpReqType> &req,
const std::vector<TBlob> &out_data,
const std::vector<TBlob> &aux_args) The output of our operators are always array of tblobs. But for most of the operators, we just have one element in the output. |
I see. |
enable setting return type for operators in registry
Most operators returns one Symbol, but some like slice_channel returns array of Symbols. This patch enables setting return type to "Symbol" or "Symbol[]" in the registry, so that we can get the return type in MXSymbolGetAtomicSymbolInfo()