IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add missed DeformableConvolution to back transformations (#10255)

Browse Source
This commit is contained in:
Anton Chetverikov 2022-02-10 17:20:11 +03:00 committed by GitHub
parent 87f8ff5918
commit 50dffb80bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/mo/openvino/tools/mo/back/blob_normalizer.py
View File

@ -33,7 +33,7 @@ class BlobNormalizer(BackReplacementPattern):
@staticmethod @staticmethod
def pattern(): def pattern():
return dict( 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=[] edges=[]
) )

1
tools/mo/openvino/tools/mo/back/op_versioning.py
View File

@ -33,6 +33,7 @@ class OpVersioning(BackReplacementPattern):
"ConvolutionBackpropData", "ConvolutionBackpropData",
"Cos", "Cos",
"Cosh", "Cosh",
'DeformableConvolution',
"DeformablePSROIPooling", "DeformablePSROIPooling",
"DepthToSpace", "DepthToSpace",
"DetectionOutput", "DetectionOutput",