Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

TODO list for supporting dynamic shape #12732

Open
zheng-da opened this issue Oct 3, 2018 · 3 comments
Open

TODO list for supporting dynamic shape #12732

zheng-da opened this issue Oct 3, 2018 · 3 comments
Labels
Backend Issues related to the backend of MXNet Feature request Operator Roadmap

Comments

@zheng-da
Copy link
Contributor

zheng-da commented Oct 3, 2018

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: [WIP] support dynamic shape in the imperative mode. #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.
@vrakesh
Copy link
Contributor

vrakesh commented Oct 3, 2018

@zheng-da thanks for the detailed proposal on supporting dynamic shapes

@mxnet-label-bot [Feature request]

@vrakesh
Copy link
Contributor

vrakesh commented Oct 3, 2018

@mxnet-label-bot [Operator]

@chinakook
Copy link
Contributor

chinakook commented Jan 11, 2019

I think that a good solution to support pseudo-dynamic (in most cases it's enough) shape is to support infer_shape_partial (through defer shape inference) in HybridBlock.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Backend Issues related to the backend of MXNet Feature request Operator Roadmap
Projects
None yet
Development

No branches or pull requests

5 participants