MXNet doesn't support operators that don't allow static shape inference, such as unique and boolean_index. One possible solution is proposed in https://cwiki.apache.org/confluence/display/MXNET/Dynamic+shape
This task can be broken into multiple steps: * support such an operator in the imperative execution. A possible modification in the imperative execution is shown here: https://github.com/apache/incubator-mxnet/pull/12400 * support such an operator in CachedOp. This step requires writing a new executor that runs operators in a computation graph one by one without shape inference and memory planning. We can start with executing operators synchronously, but eventually, we should execute them asynchronously. * an optimization is to use the subgraph API to split the computation graph into subgraphs so that a subgraph either contain only operators allowing static shape inference or operators with dynamic shape. [ Full content available at: https://github.com/apache/incubator-mxnet/issues/12732 ] This message was relayed via gitbox.apache.org for [email protected]
