Skip to content
This repository was archived by the owner on Aug 30, 2018. It is now read-only.

Commit 13816aa

Browse files
committed
Add support for Unsqueeze
1 parent 59a0ae3 commit 13816aa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnx_caffe2/backend.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,11 +141,13 @@ class Caffe2Backend(Backend):
141141
'Equal': 'EQ',
142142
'Less': 'LT',
143143
'Greater': 'GT',
144+
'Unsqueeze': 'ExpandDims',
144145
}
145146

146147
_global_renamed_attrs = {'kernel_shape': 'kernels'}
147148
_per_op_renamed_attrs = {
148149
'Squeeze': {'axes': 'dims'},
150+
'Unsqueeze': {'axes': 'dims'},
149151
'Transpose': {'perm': 'axes'},
150152
'Upsample': {'mode': ''},
151153
'ConvTranspose': {'output_padding': 'adjs'},

0 commit comments

Comments
 (0)