tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From jroe...@apache.org
Subject [incubator-tvm] branch master updated (a6de507 -> e6dd8e1)
Date Tue, 24 Mar 2020 06:56:39 GMT
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


Mime
View raw message