Python Add.update_node_stat Examples

Example #1

    def extract(node: Node):
        attrs = {'axis': node.pb.bias_param.axis}
        embed_input(attrs, 1, 'bias', node.model_pb.blobs[0].data, 'biases')

        Add.update_node_stat(node, attrs)

        return __class__.enabled

Example #2

 def extract(node):
     axis = onnx_attr(node, 'axis', 'i', default=None)
     Add.update_node_stat(node, {'axis': axis})
     return __class__.enabled