Fea/nn graph/block proxy func (#5727)
* pass test on linear with training * Refactor RuntimeCtx for multi-runtime * refactor inplace to support nn graph * block support iterator * block iter add check * fix scalar_mul op conf build * deal with inplace after merge master * add alexnet graph test * add cpu test and format * cout to glog * deal with Job run finish bug * refactor lazy deal with inplace * deal with 0D tensor * update data path * address review * deal with lazy default attr * mv according to ci * merge master * fix for ci * fix for ci limit * block proxy func * support module custom func and refacotr get attr of block * auto format by CI Co-authored-by:chengtbf <472491134@qq.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- python/oneflow/nn/graph_block.py 45 additions, 40 deletionspython/oneflow/nn/graph_block.py
- python/oneflow/test/graph/test_graph.py 4 additions, 2 deletionspython/oneflow/test/graph/test_graph.py
- python/oneflow/test/graph/test_graph_block.py 109 additions, 0 deletionspython/oneflow/test/graph/test_graph_block.py
Please register or sign in to comment