36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
# Copyright (C) 2018-2021 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from mo.back.replacement import BackReplacementPattern
|
|
from mo.graph.graph import Node, Graph
|
|
from mo.utils.error import Error
|
|
|
|
|
|
class DisableUnsupportedNDOperations(BackReplacementPattern):
|
|
"""
|
|
This pass disables ND Convolutions/Deconvolutions/Poolings
|
|
"""
|
|
enabled = False
|
|
|
|
unsupported_operations = ['Convolution', 'Deconvolution', 'Pooling']
|
|
|
|
def find_and_replace_pattern(self, graph: Graph):
|
|
unsupported_nodes = []
|
|
for node in graph.nodes():
|
|
node = Node(graph, node)
|
|
if node.kind == 'op' and node.soft_get('type') in self.unsupported_operations:
|
|
input_shape = node.in_node(0).shape
|
|
if len(input_shape) > 4:
|
|
unsupported_nodes.append((node.id, node.type))
|
|
|
|
if len(unsupported_nodes) == 0:
|
|
return
|
|
|
|
error_message = "\nOperations below were marked as unsupported due to they expect more than two spatial dims" \
|
|
" (input shape length more than 4)\n"
|
|
error_message += "List of unsupported operations ({})\n".format(len(unsupported_nodes))
|
|
for node, type in unsupported_nodes:
|
|
error_message += " {} {}\n".format(type, node)
|
|
|
|
raise Error(error_message)
|