Skip to content

Commit

Permalink
test fix
Browse files Browse the repository at this point in the history
  • Loading branch information
p-wysocki committed May 22, 2024
1 parent 72ad8b1 commit 353a39c
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions src/bindings/python/src/openvino/runtime/opset15/ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,12 +65,14 @@ def col2im(
:return: The new node performing Col2Im operation.
"""
for param in [strides, dilations]:
if param is None:
param = [1, 1]
for param in [pads_begin, pads_end]:
if param is None:
param = [0, 0]
if strides is None:
strides = [1, 1]
if dilations is None:
dilations = [1, 1]
if pads_begin is None:
pads_begin = [0, 0]
if pads_end is None:
pads_end = [0, 0]
return _get_node_factory_opset15().create(
"Col2Im",
as_nodes(data, output_size, kernel_size, name=name),
Expand Down

0 comments on commit 353a39c

Please sign in to comment.