Unverified Commit 22286f71 authored by mvermeulen's avatar mvermeulen Committed by GitHub
Browse files

Merge branch 'develop' into batchnorm-rewrite

parents 4f447b03 da782877
...@@ -253,6 +253,12 @@ struct miopen_conv_bias ...@@ -253,6 +253,12 @@ struct miopen_conv_bias
fusion::op_t conv; fusion::op_t conv;
fusion::op_t bias; fusion::op_t bias;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return op::convolution::reflect(self.op, f);
}
miopen_conv_bias(op::convolution c, const shape& input, const shape& weights, const shape& b) miopen_conv_bias(op::convolution c, const shape& input, const shape& weights, const shape& b)
: op(c), f(input) : op(c), f(input)
{ {
...@@ -290,6 +296,12 @@ struct miopen_conv_bias_relu ...@@ -290,6 +296,12 @@ struct miopen_conv_bias_relu
fusion::op_t bias; fusion::op_t bias;
fusion::op_t relu; fusion::op_t relu;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return op::convolution::reflect(self.op, f);
}
miopen_conv_bias_relu(op::convolution c, miopen_conv_bias_relu(op::convolution c,
const shape& input, const shape& input,
const shape& weights, const shape& weights,
......
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