Add missed DeformableConvolution to back transformations (#10255)

This commit is contained in:
Anton Chetverikov authored and GitHub committed 2022-02-10 17:20:11 +03:00
1 parent 87f8ff5918
commit 50dffb80bb
2 files changed
+2 -1

No files matched your search

@@ -33,7 +33,7 @@ class BlobNormalizer(BackReplacementPattern):
@staticmethod
def pattern():
return dict(
nodes=[('conv', dict(type=lambda type: type in ['Convolution', 'Deconvolution', 'FullyConnected']))],
nodes=[('conv', dict(type=lambda type: type in ['Convolution', 'Deconvolution', 'FullyConnected', 'DeformableConvolution']))],
edges=[]
)
@@ -33,6 +33,7 @@ class OpVersioning(BackReplacementPattern):
"ConvolutionBackpropData",
"Cos",
"Cosh",
'DeformableConvolution',
"DeformablePSROIPooling",
"DepthToSpace",
"DetectionOutput",