[Relay] GradientCell Relay Pass (#5039)
authorAndrew Liu <andrewlliu@gmail.com>
Tue, 24 Mar 2020 06:56:28 +0000 (23:56 -0700)
committerGitHub <noreply@github.com>
Tue, 24 Mar 2020 06:56:28 +0000 (23:56 -0700)
commite6dd8e1e7b38fc2862c53f445899997bb8a19534
tree15ee2b2e50a59190b6572cfbdf7912cd485d3256
parenta6de507b635df1df3753c0f6d14b0f0e216fcf4a
[Relay] GradientCell Relay Pass (#5039)

* save

* gradient.rly

* fix

* NOT WORKING: gradient cell pass

* test gradient pass

* fixed basic call ops

* more tests

* fix bug

* transform calls to one ones_like zero zero_like

* maintenance stuff

* fix linting

* linting

* linting

* throw default

* remove unrelated changes

* import gradent.rly in pass

* comment

* linting

* remove changes to test files

* move gradient_cell.cc to transforms

* revert change

* update files with new commits

* type

* wrapper function to main outermost function type

* fix linting

* fix unsigned and signed int comparison

* review

* GetConstructor definition in module and change op comparison

* update node instantiations

* increase code readability

Co-authored-by: Marisa Kirisame <lolisa@marisa.moe>
include/tvm/ir/module.h
include/tvm/relay/transform.h
python/tvm/relay/std/gradient.rly [new file with mode: 0644]
python/tvm/relay/transform/transform.py
src/ir/module.cc
src/relay/transforms/lazy_gradient_init.cc [new file with mode: 0644]
tests/python/relay/test_ir_parser.py
tests/python/relay/test_pass_lazy_gradient_init.py [new file with mode: 0644]