Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim.#12971
Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim.#12971gujinghui wants to merge 1 commit intopytorch:masterfrom
Conversation
|
@pytorchbot retest this please |
yinghai
left a comment
There was a problem hiding this comment.
The title of the PR is misleading. For general FC op, we already support it.
This is about Nomnigraph transformation and support in IDEEP, right?
|
@yinghai Yes, the title is misleading. Will change it. |
6373512 to
b975ae3
Compare
facebook-github-bot
left a comment
There was a problem hiding this comment.
yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
b975ae3 to
334dac0
Compare
|
@yinghai |
|
@yinghai |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
yinghai
left a comment
There was a problem hiding this comment.
I think this needs rebase.
334dac0 to
13f52f7
Compare
|
@yinghai |
|
Could you fix the clang format too? I found that your style seems to be consistently different from ours. :0 |
13f52f7 to
55b7b1f
Compare
Sure. Done. Pls help review. |
75311c4 to
bb50cdc
Compare
|
rebased to latest. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@gujinghui Do we have any tests for this change? |
bb50cdc to
5f5ed84
Compare
|
@yinghai rebased to latest. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
caffe2/core/nomnigraph/include/nomnigraph/Generated/OpClasses.h
Outdated
Show resolved
Hide resolved
Signed-off-by: Gu, Jinghui <jinghui.gu@intel.com>
5f5ed84 to
8a95554
Compare
|
@yinghai |
facebook-github-bot
left a comment
There was a problem hiding this comment.
@yinghai has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Add "axis" and "axis_w" arguments in FC to support customized axix to reduce dim.