File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
model_compression_toolkit/target_platform_capabilities/targetplatform2framework Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 3232from model_compression_toolkit .target_platform_capabilities .targetplatform2framework .attach2fw import \
3333 AttachTpcToFramework
3434from model_compression_toolkit .target_platform_capabilities .targetplatform2framework .attribute_filter import Eq
35- from sony_custom_layers .pytorch import MulticlassNMS , MulticlassNMSWithIndices , FasterRCNNBoxDecode
3635
3736
3837class AttachTpcToPytorch (AttachTpcToFramework ):
@@ -98,8 +97,8 @@ def __init__(self):
9897 OperatorSetNames .L2NORM : [LayerFilterParams (torch .nn .functional .normalize ,
9998 Eq ('p' , 2 ) | Eq ('p' , None ))],
10099 OperatorSetNames .SSD_POST_PROCESS : [], # no such operator in pytorch
101- OperatorSetNames .COMBINED_NON_MAX_SUPPRESSION : [MulticlassNMS , MulticlassNMSWithIndices ], # no such operator in pytorch
102- OperatorSetNames .BOX_DECODE : [FasterRCNNBoxDecode ]
100+ OperatorSetNames .COMBINED_NON_MAX_SUPPRESSION : [], # no such operator in pytorch
101+ OperatorSetNames .BOX_DECODE : [] # no such operator in pytorch
103102 }
104103
105104 pytorch_linear_attr_mapping = {KERNEL_ATTR : DefaultDict (default_value = PYTORCH_KERNEL ),
You can’t perform that action at this time.
0 commit comments