This is an automated email from the ASF dual-hosted git repository.
jroesch pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from a6de507 Change Rust version to stable in Docker (#5138)
add e6dd8e1 [Relay] GradientCell Relay Pass (#5039)
No new revisions were added by this update.
Summary of changes:
include/tvm/ir/module.h | 8 +
include/tvm/relay/transform.h | 14 +
.../tvm/relay/std/gradient.rly | 48 ++-
python/tvm/relay/transform/transform.py | 13 +
src/ir/module.cc | 12 +
src/relay/transforms/lazy_gradient_init.cc | 317 +++++++++++++++++
tests/python/relay/test_ir_parser.py | 5 +-
tests/python/relay/test_pass_lazy_gradient_init.py | 395 +++++++++++++++++++++
8 files changed, 794 insertions(+), 18 deletions(-)
copy tests/cpp/simple_passes_test.cc => python/tvm/relay/std/gradient.rly (50%)
create mode 100644 src/relay/transforms/lazy_gradient_init.cc
create mode 100644 tests/python/relay/test_pass_lazy_gradient_init.py
