Skip to content

Commit

Permalink
Fix opset version collisions
Browse files Browse the repository at this point in the history
  • Loading branch information
sovrasov committed Jun 19, 2023
1 parent 91507bc commit 767a8cc
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion otx/core/ov/ops/activations.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class SoftMaxV1(Operation[SoftMaxV1Attribute]):
"""SoftMaxV1 class."""

TYPE = "Softmax"
VERSION = "opset1"
VERSION = "opset2"
ATTRIBUTE_FACTORY = SoftMaxV1Attribute

def forward(self, inputs):
Expand Down
2 changes: 1 addition & 1 deletion otx/core/ov/ops/movements.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ class GatherV1(Operation[GatherV1Attribute]):
"""GatherV1 class."""

TYPE = "Gather"
VERSION = "opset1"
VERSION = "opset2"
ATTRIBUTE_FACTORY = GatherV1Attribute

def forward(self, inputs, indices, axis):
Expand Down
2 changes: 1 addition & 1 deletion otx/core/ov/ops/shape_manipulations.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ class ShapeOfV3(Operation[ShapeOfV3Attribute]):
"""ShapeOfV3 class."""

TYPE = "ShapeOf"
VERSION = "opset1"
VERSION = "opset3"
ATTRIBUTE_FACTORY = ShapeOfV3Attribute

def forward(self, inputs):
Expand Down
2 changes: 1 addition & 1 deletion otx/core/ov/ops/sorting_maximization.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ class NonMaxSuppressionV5(Operation[NonMaxSuppressionV5Attribute]):
"""NonMaxSuppressionV5 class."""

TYPE = "NonMaxSuppression"
VERSION = "opset1"
VERSION = "opset5"
ATTRIBUTE_FACTORY = NonMaxSuppressionV5Attribute

def forward(
Expand Down

0 comments on commit 767a8cc

Please sign in to comment.