ONE
ONE copied to clipboard
[circle-partitioner] Support partition by node index
Let's support partition model by node index.
currently support is by operator code name and node name.
Idea from https://github.com/Samsung/ONE/issues/7596#issuecomment-911865240
-
1,2,3
-> 1, 2, 3 -
1,3-4
-> 1, 3, 4 -
1, 4-
-> 1, 4, 5, 6, ..., N -
-3, 7
-> 1, 2, 3, 7
Support this for now, and other notations can be extended later.
@seanshpark Do you have any idea how to identify node index with multiple subgraph ? (AFAIK, each subgraph has its own indices starting at 0 )
Do you have any idea how to identify node index with multiple subgraph ?
As of now, partition is done for first(main) sub-graph. other sub-graphs follow it's owner.
Giving index, name for other sub-graphs is currently a debt for future me(or who ever is going to do)... -_-;
Status
-
by node index
is about Tensor index incircle
(ortflite
) file - circle IR doesn't store Tensor index, it's generated when export is done
For partitioner(or other tools), how to identify circle Node from Tensor index?
- TBD