Skip to content

Conversation

@mbrookhart
Copy link
Contributor

"NCHW",
"NHWC",
"NDHWC",
], "Layout must be one of 'NWC', 'NCHW', NHWC or NDHWC"
Copy link
Contributor

Choose a reason for hiding this comment

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

looking at this made me realize we should just have 'channels_first' and 'channels_last' as layout options so that it applies to all dimensions. No change needed for this PR but we should keep it in mind as a backlog item to fix.

Copy link
Contributor

@jwfromm jwfromm left a comment

Choose a reason for hiding this comment

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

LGTM!

@masahi masahi merged commit 0d46cf7 into apache:main Dec 4, 2020
@masahi
Copy link
Member

masahi commented Dec 4, 2020

Thanks @mbrookhart @jwfromm

TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Jan 20, 2021
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Jan 21, 2021
electriclilies pushed a commit to electriclilies/tvm that referenced this pull request Feb 18, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants