Commit a9c0252a authored by charlie's avatar charlie
Browse files

Revert ref_conv changes

Not needed, special case with dynamic padding
parent 0a9c10b3
...@@ -232,9 +232,8 @@ struct ref_convolution : auto_register_op<ref_convolution<Op>> ...@@ -232,9 +232,8 @@ struct ref_convolution : auto_register_op<ref_convolution<Op>>
return op.normalize_compute_shape(inputs); return op.normalize_compute_shape(inputs);
} }
argument compute(context&, migraphx::dyn_output dyn_output, std::vector<argument> args) const argument compute(context&, shape output_shape, std::vector<argument> args) const
{ {
shape output_shape;
std::vector<std::size_t> padding; std::vector<std::size_t> padding;
if(op.padding_mode != op::padding_mode_t::default_) if(op.padding_mode != op::padding_mode_t::default_)
{ {
...@@ -249,8 +248,12 @@ struct ref_convolution : auto_register_op<ref_convolution<Op>> ...@@ -249,8 +248,12 @@ struct ref_convolution : auto_register_op<ref_convolution<Op>>
} }
else else
{ {
padding = op.padding; padding = op.padding;
output_shape = dyn_output.computed_shape; if(output_shape.dynamic())
{
output_shape =
op.normalize_compute_shape({args.at(0).get_shape(), args.at(1).get_shape()});
}
} }
argument result{output_shape}; argument result{output_shape};
......
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