This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from ff7bab8 [TOPI] Setting workload correctly for Depthwise conv ARM.
(#5182)
add e4a5441 [TE] reverse-mode autodiff without any optimization (#5121)
No new revisions were added by this update.
Summary of changes:
include/tvm/te/autodiff.h | 97 ++++++
python/tvm/te/__init__.py | 1 +
python/tvm/te/autodiff.py | 67 ++++
src/te/autodiff/ad_util.cc | 64 ++++
.../aocl/aocl_module.h => te/autodiff/ad_util.h} | 42 +--
src/te/autodiff/adjoint.cc | 155 +++++++++
src/te/autodiff/jacobian.cc | 378 +++++++++++++++++++++
src/te/operation/compute_op.h | 1 -
tests/python/unittest/test_te_autodiff.py | 172 ++++++++++
tests/python/unittest/test_testing.py | 2 +
10 files changed, 958 insertions(+), 21 deletions(-)
create mode 100644 include/tvm/te/autodiff.h
create mode 100644 python/tvm/te/autodiff.py
create mode 100644 src/te/autodiff/ad_util.cc
copy src/{runtime/opencl/aocl/aocl_module.h => te/autodiff/ad_util.h} (52%)
create mode 100644 src/te/autodiff/adjoint.cc
create mode 100644 src/te/autodiff/jacobian.cc
create mode 100644 tests/python/unittest/test_te_autodiff.py
