Skip to content

[Relay][Training] Add and fix gradients - #4126

Merged
vinx13 merged 2 commits into
apache:masterfrom
altanh:grads
Oct 16, 2019
Merged

[Relay][Training] Add and fix gradients#4126
vinx13 merged 2 commits into
apache:masterfrom
altanh:grads

Conversation

@altanh

Copy link
Copy Markdown
Contributor

Added the following gradient:

  • nn.global_avg_pool2d

Fixed the following gradients:

  • max
  • sum

New regression tests were added as well.

Many thanks to @Checkmate50@dfujiki@xiaoweiw for contributing to this PR at the ADA TVM hackathon!

@altanh

Copy link
Copy Markdown
ContributorAuthor

cc @MarisaKirisame

@MarisaKirisameMarisaKirisame left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@vinx13
vinx13 merged commit 46fa6ee into apache:masterOct 16, 2019
anijain2305 pushed a commit to anijain2305/tvm that referenced this pull request Oct 17, 2019
* add and fix gradients
* fix linter issues
wweic pushed a commit to neo-ai/tvm that referenced this pull request Oct 18, 2019
* add and fix gradients
* fix linter issues
petrex pushed a commit to petrex/tvm that referenced this pull request Oct 29, 2019
* master: (51 commits)
[QNN][TFLite] Parsing QNN Add op. Adding MobilenetV2. (apache#4142)
[CI] Pin NNPack pthreadtools version (apache#4152)
Fix typo (apache#4144)
[Relay][Frontend][TF] Add tensor array ops (apache#3798)
[relay][vm] Separate VM runtime with executable (apache#4100)
[PATCH] Fix undefined __floatdihf in libtvmruntime.so on aarch64. (apache#4119)
[DOCKER] Pin torchvision==0.4.1 (apache#4140)
[TOPI][x86] Cascade lake support. (apache#4123)
[Relay] Improve build error when no lowered funcs are produced (apache#4132)
[RUNTIME] Refactor object python FFI to new protocol. (apache#4128)
Update PULL_REQUEST_TEMPLATE.md
Adding support for dequantizing from int32 to float32. (apache#4130)
[Relay][Training] Add and fix gradients (apache#4126)
[QNN] Change default rouning to UPWARD. (apache#4131)
Fix infer type of kernel in dense. (apache#4125)
[Relay][AlterOpLayout] NHWC to NCHWc pad operator. (apache#4103)
[ARITH] Fix lowering of floormod(x, y) != 0 (apache#4127)
[RFC][RUNTIME] Introduce new object protocol. (apache#4115)
[Relay][Topi] Disable conv NHWC pack int8. (apache#4038)
Update task_cpp_unittest.sh
...
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

@altanh@MarisaKirisame@vinx13