def extract(cls, 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 cls.enabled
def extract(cls, node: Node): axis = onnx_attr(node, 'axis', 'i', default=None) Add.update_node_stat(node, {'axis': axis}) return cls.enabled
def extract(cls, node): Add.update_node_stat(node, {'data_type': tf_dtype_extractor(node.pb.attr["T"].type)}) return cls.enabled
def extract(cls, node): Add.update_node_stat(node, {}) return cls.enabled