As GraphRuntime does not provide control-flow logics, we have to split our model to two parts. While we need to share parameters between them to save memory usage. Solution: 1) add "lazy_init_input" in graph's attributes "attrs": { ... ... "lazy_init_input": [ "list_str", [ "p0" ] ] } 2) allow un-allocated NDArray entry in SetupStorage 3) utilize "set_input_zero_copy" function to set parameters
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
.gitignore | Loading commit data... | |
attrs_test.cc | Loading commit data... | |
build_module_test.cc | Loading commit data... | |
container_test.cc | Loading commit data... | |
expr_test.cc | Loading commit data... | |
ir_functor_test.cc | Loading commit data... | |
ir_mutator_test.cc | Loading commit data... | |
ir_simplify_test.cc | Loading commit data... | |
ir_ssa_test.cc | Loading commit data... | |
ir_visitor_test.cc | Loading commit data... | |
packed_func_test.cc | Loading commit data... | |
pattern_match_test.cc | Loading commit data... | |
relay_build_module_test.cc | Loading commit data... | |
relay_pass_type_infer_test.cc | Loading commit data... | |
relay_transform_sequential.cc | Loading commit data... | |
simple_passes_test.cc | Loading commit data... | |
tensor_test.cc | Loading commit data... | |
threading_backend_test.cc | Loading commit data... | |
topi_ewise_test.cc | Loading commit data... |