Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 3bc04209 编写于 作者: L liuqi
浏览文件
操作

Add default value to dilation argument of conv2d.

上级 51f366fb
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mace/ops/conv_pool_2d_base.h
浏览文件 @ 3bc04209
...@@ -18,7 +18,7 @@ class ConvPool2dOpBase : public Operator<D, T> { ...@@ -18,7 +18,7 @@ class ConvPool2dOpBase : public Operator<D, T> {
strides_(OperatorBase::GetRepeatedArgument<int>("strides")), strides_(OperatorBase::GetRepeatedArgument<int>("strides")),
padding_(static_cast<Padding>(OperatorBase::GetSingleArgument<int>( padding_(static_cast<Padding>(OperatorBase::GetSingleArgument<int>(
"padding", static_cast<int>(SAME)))), "padding", static_cast<int>(SAME)))),
dilations_(OperatorBase::GetRepeatedArgument<int>("dilations")) {} dilations_(OperatorBase::GetRepeatedArgument<int>("dilations", {1, 1})) {}
void CalOutputSize(const index_t *input_shape, // NCHW void CalOutputSize(const index_t *input_shape, // NCHW
const index_t *filter_shape, // OIHW const index_t *filter_shape, // OIHW
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或