-
Notifications
You must be signed in to change notification settings - Fork 16
Sarkars/unify pooling translate functions #420
base: master
Are you sure you want to change the base?
Sarkars/unify pooling translate functions #420
Conversation
std::shared_ptr<ng::Node> ng_avgpool = | ||
make_shared<ng::op::AvgPool>(ng_input, ng_kernel_shape, ng_strides, | ||
ng_padding_below, ng_padding_above, false); | ||
std::shared_ptr<ng::Node> ng_pool = |
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.
Question:
Does TF AvgPool considers padding in average calculation? Just want to confirm here.
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.
Avg pool has this extra param whose default value is false:
https://github.com/NervanaSystems/ngraph/blob/066037c25901478ac79ead90aa6da721ab27c0c3/src/ngraph/op/avg_pool.hpp#L51
Earlier we were explicitly passing the false
value, but even without it, the default value is going to be false
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.
Cool! Sounds good!
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!
e696e3d
to
7978347
Compare
No description provided.