-
Notifications
You must be signed in to change notification settings - Fork 157
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
[cker] Introduce cker for avgpool #14086
Conversation
9007319
to
ca8adda
Compare
This commit add chekr for average pooling ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
ca8adda
to
3f3785e
Compare
out_mat.cwiseMin(params.float_activation_min).cwiseMax(params.float_activation_max); | ||
} | ||
|
||
inline void AvgPool2DGrad(const PoolParams ¶ms, const Shape &incoming_shape, |
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.
Could you add the backward formula for avgpool, here? #11248
AvgPoolOpVerifier<float> verifier(op_param, in, out); | ||
|
||
/** | ||
* input(index) : output(arg-count): |
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.
(optional)
Ah, If you followed the maxpool2d
notation, maxpool2d
expressed the index - because one output of maxpool2d
is about index
.
For avergepool2d, index isn't important, I guess you don't need to mark the index here.
If you have another intention about index
plz let me know.
AvgPoolOpVerifier<float> verifier(op_param, in, out); | ||
|
||
/** | ||
* input(index) : output(arg-count): |
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.
* input(index) : output(arg-count): | |
* input : output(arg-count): |
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.
Ah, If you followed the
maxpool2d
notation
I did follow the maxpool2d
notation. I'll remove it.
use existing forwarding function. update notaion. ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
I update it! |
* limitations under the License. | ||
*/ | ||
|
||
#ifndef __NNFW_CKER_TRAIN_OPERATION_AVGPOOL_H__ |
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.
#ifndef __NNFW_CKER_TRAIN_OPERATION_AVGPOOL_H__ | |
#ifndef __NNFW_CKER_TRAIN_OPERATION_AVERAGEPOOL_H__ |
Same to the following two occurrences
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.
Updated!
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.
LGTM
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.
LGTM
This commit updates header guard. ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
This PR introduce cker for average pooling operation.
ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]