-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Relay][TOPI][OP] intel_graphics conv2d alterlayout support relay, added stack op #2729
Conversation
if F == sym: | ||
out = F.contrib.conv2d_NCHWc(*copy_inputs, **new_attrs) | ||
else: | ||
out = F.nn.contrib_conv2d_nchwc(*copy_inputs, **new_attrs) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
does it work? Conv2dAttr
contains layout
in nnvm while in Relay the name is data_layout
though
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@yzhliu It seems working for me. relay.nn.contrib_conv2d_nchwc
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
oh yes, because it is copy, relay copies to relay, nnvm copies to nnvm.
Thanks @Laurawly |
…ded stack op (apache#2729) * add stack op frontend * concate moved * topi stack added * stack added * fix stack bugs and tested * conv2d alterlayout udpated for relay * fix pylint * fix cmake warnings * cmake warnings fixed
…ded stack op (apache#2729) * add stack op frontend * concate moved * topi stack added * stack added * fix stack bugs and tested * conv2d alterlayout udpated for relay * fix pylint * fix cmake warnings * cmake warnings fixed
…ded stack op (apache#2729) * add stack op frontend * concate moved * topi stack added * stack added * fix stack bugs and tested * conv2d alterlayout udpated for relay * fix pylint * fix cmake warnings * cmake warnings fixed
…ded stack op (apache#2729) * add stack op frontend * concate moved * topi stack added * stack added * fix stack bugs and tested * conv2d alterlayout udpated for relay * fix pylint * fix cmake warnings * cmake warnings fixed
Also concatenate op moved compute to align with other ops.