提交 aad11531 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Also fix MaxPoolRop

上级 ded3fd27
......@@ -106,7 +106,7 @@ KERNEL void max_pool3d_rop_kernel(const ga_size nthreads,
#define OUTPUT_DIMS(in_dim, ws, st) \
(IGNORE_BORDER ? (in_dim - ws)/st + 1 : \
(st > ws ? (in_dim - 1)/st + 1 : \
std::max<size_t>(0, (in_dim - 1 - ws + st)/st) + 1))
std::max<ssize_t>(0, (in_dim - 1 - ws + st)/st) + 1))
#section support_code_struct
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论