Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
af974c34

Unverified Commit af974c34 authored 6 years ago by Tianqi Chen Committed by GitHub 6 years ago

Downloads
- Patches
- Plain Diff

[RELAY][OP] Fix conv2d NHWC type inference. (#2019)

parent 42dc24a3

No related branches found

No related tags found

No related merge requests found

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/packed_func_ext.h 5 additions, 3 deletions

include/tvm/packed_func_ext.h
src/relay/ir/op.cc 5 additions, 0 deletions

src/relay/ir/op.cc
src/relay/op/nn/convolution.cc 12 additions, 8 deletions

src/relay/op/nn/convolution.cc
src/relay/op/nn/layout.h 1 addition, 3 deletions

src/relay/op/nn/layout.h
tests/python/relay/test_op_level2.py 18 additions, 3 deletions

tests/python/relay/test_op_level2.py

with 41 additions and 17 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment