Python ExpandDims.update_node_stat Examples

Example #1

File: unsqueeze_ext.py Project: shyransystems/OpenVINO-toolkit-orig

    def extract(cls, node):
        axis = np.array(onnx_attr(node, 'axes', 'ints', default=[]),
                        dtype=np.int64)

        ExpandDims.update_node_stat(node, {'expand_axis': axis})
        return cls.enabled

Example #2

File: expand_dims_ext.py Project: zhenlusu500/openvino

 def extract(cls, node):
     attrs = get_mxnet_layer_attrs(node.symbol_dict)
     expand_axis = attrs.int('axis', None)
     ExpandDims.update_node_stat(node, {'expand_axis': expand_axis})
     return cls.enabled