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

Merge pull request #220 from ROCmSoftwarePlatform/show-conv-fusion-info

Show convolution settings on fusions
parents db70de8e 75c06bd2
...@@ -251,6 +251,12 @@ struct miopen_conv_bias ...@@ -251,6 +251,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)
{ {
...@@ -288,6 +294,12 @@ struct miopen_conv_bias_relu ...@@ -288,6 +294,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