Exemplos de ExpandDims.update_node_stat em Python

Exemplo n.º 1

Arquivo: unsqueeze_ext.py Projeto: 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

Exemplo n.º 2

Arquivo: expand_dims_ext.py Projeto: 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