Commit 8276bc83 authored by Khalique's avatar Khalique
Browse files

formatting

parent e17fb984
...@@ -63,7 +63,7 @@ struct convolution ...@@ -63,7 +63,7 @@ struct convolution
valid valid
}; };
padding_mode_t padding_mode = default_; padding_mode_t padding_mode = default_;
int group = 1; int group = 1;
template <class Self, class F> template <class Self, class F>
static auto reflect(Self& self, F f) static auto reflect(Self& self, F f)
......
...@@ -55,7 +55,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op) ...@@ -55,7 +55,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op)
{ {
auto c = make_obj<convolution_descriptor>(&miopenCreateConvolutionDescriptor); auto c = make_obj<convolution_descriptor>(&miopenCreateConvolutionDescriptor);
miopenConvolutionMode_t c_mode = miopenConvolution; miopenConvolutionMode_t c_mode = miopenConvolution;
if (op.group > 1) if(op.group > 1)
c_mode = miopenGroupConv; c_mode = miopenGroupConv;
miopenInitConvolutionDescriptor(c.get(), miopenInitConvolutionDescriptor(c.get(),
c_mode, c_mode,
...@@ -65,7 +65,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op) ...@@ -65,7 +65,7 @@ inline convolution_descriptor make_conv(const migraphx::op::convolution& op)
op.stride[1], op.stride[1],
op.dilation[0], op.dilation[0],
op.dilation[1]); op.dilation[1]);
if (op.group > 1) if(op.group > 1)
miopenSetConvolutionGroupCount(c.get(), op.group); miopenSetConvolutionGroupCount(c.get(), op.group);
return c; return c;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment