Commit f8d4daa2 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU Direct Execution: Migrate all convolution variants to auto-registration

parent 9376bd0c
...@@ -403,6 +403,14 @@ namespace ngraph ...@@ -403,6 +403,14 @@ namespace ngraph
"ConvolutionBiasBackpropFiltersBias is only supported with MKLDNN kernel."); "ConvolutionBiasBackpropFiltersBias is only supported with MKLDNN kernel.");
} }
} }
REGISTER_OP_BUILDER(Convolution);
REGISTER_OP_BUILDER(ConvolutionRelu);
REGISTER_OP_BUILDER(ConvolutionBias);
REGISTER_OP_BUILDER(ConvolutionBiasAdd);
REGISTER_OP_BUILDER(ConvolutionBackpropData);
REGISTER_OP_BUILDER(ConvolutionBackpropFilters);
REGISTER_OP_BUILDER(ConvolutionBiasBackpropFiltersBias);
} }
} }
} }
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