[MXNET-378] Adding depth_to_space and space_to_depth operator(Updated)#11587
Conversation
|
@piiswrong @reminisce @anirudh2290 @eric-haibin-lin @rahul003 Please help with review when you have time, thanks! |
| NNVM_REGISTER_OP(_backward_squeeze) | ||
| .set_attr<FCompute>("FCompute<gpu>", UnaryOp::IdentityCompute<gpu>); | ||
|
|
||
| NNVM_REGISTER_OP(depth_to_space) |
There was a problem hiding this comment.
Please also update https://github.com/apache/incubator-mxnet/blob/master/docs/api/python/ndarray/ndarray.md and docs/api/python/symbol/symbol.md when adding new ops
anirudh2290
left a comment
There was a problem hiding this comment.
Thank you for your contribution! Good work.
ce29f04 to
0187ca1
Compare
haojin2
left a comment
There was a problem hiding this comment.
In general LGTM but please introduce the randomized tests.
|
Agree with @haojin2 would be good to have some random tests |
eric-haibin-lin
left a comment
There was a problem hiding this comment.
The link doesn't look right.
http://mxnet-ci-doc.s3-accelerate.dualstack.amazonaws.com/PR-11587/14/api/python/ndarray/ndarray.html#array-rearrange-elements
The document is not very readable and the format seems off.
http://mxnet-ci-doc.s3-accelerate.dualstack.amazonaws.com/PR-11587/14/api/python/ndarray/ndarray.html#mxnet.ndarray.depth_to_space
… added block_size check
apache#11587) * [MXNET-378] Adding depth_to_space and space_to_depth operator * fixed lint and windows CPU errors * compliance with C++ style guiide and address shortcomings in unittests * fixed documentation and nitpicky suggestions * added operator references in API docs and removed inplace optimization support * Added references in symbol.md and ndarray.md. Improved test cases and added block_size check * Fixing bugs in documentation. Tests now include tensors of random shapes.
Description
Added new operators "depth_to_space" and "space_to_depth" to mxnet
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Comments
Ran Tests on CPU and GPU for both depth_to_space and space_to_depth operator 10000 times