projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 5d445ca) | patch
[RELAY][GRAD] handle Tuple/TupleGetItem in first order gradient (#5946)
authorThomas Viehmann <tv.code@beamnet.de>
Tue, 30 Jun 2020 03:34:20 +0000 (05:34 +0200)
committerGitHub <noreply@github.com>
Tue, 30 Jun 2020 03:34:20 +0000 (20:34 -0700)
commit957aefbb174749602fcb138f88bd877f6ff9cd3c
tree76a23807b6a09f911788401cd50357aa15c24980tree | snapshot
parent5d445ca4c60bcbdb02c6932eb58c9fa6abd4412acommit | diff
[RELAY][GRAD] handle Tuple/TupleGetItem in first order gradient (#5946)

* handle Tuple/TupleGetItem in first order gradient

* Unify MultiOnes/MultiZeros.
src/relay/transforms/gradient.cc diff | blob | history
src/relay/transforms/pattern_util.h diff | blob | history
tests/python/relay/test_pass_gradient.py diff | blob | history
Domain: Machine Learning / ML Framework;