Exemplos de AttributeProto em Python

Exemplo n.º 1

def _Attribute_default_value(self):
    attr = onnx.AttributeProto()
    attr.ParseFromString(self._default_value)
    return attr

Exemplo n.º 2

Arquivo: util.py Projeto: semihlab/onnx-connx

def _strings(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.STRINGS
    attr.strings.extends([str.encode(s, 'utf-8') for s in value])
    return attr

Exemplo n.º 3

Arquivo: test_attribute.py Projeto: lightmatter-ai/onnx2pytorch

def test_extract_attr_values(kwargs, value):
    attr = onnx.AttributeProto(**kwargs)
    assert extract_attr_values(attr) == value

Exemplo n.º 4

Arquivo: util.py Projeto: semihlab/onnx-connx

def _float(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.FLOAT
    attr.f = value
    return attr

Exemplo n.º 5

Arquivo: util.py Projeto: semihlab/onnx-connx

def _ints(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.INTS
    attr.ints.extend(value)
    return attr

Exemplo n.º 6

Arquivo: util.py Projeto: semihlab/onnx-connx

def _floats(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.FLOATS
    attr.floats.extend(value)
    return attr

Exemplo n.º 7

Arquivo: util.py Projeto: semihlab/onnx-connx

def _string(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.STRING
    attr.s = str.encode(value, 'utf-8')
    return attr

Exemplo n.º 8

Arquivo: util.py Projeto: semihlab/onnx-connx

def _int(name, value):
    attr = onnx.AttributeProto()
    attr.name = name
    attr.type = onnx.AttributeProto.AttributeType.INT
    attr.i = value
    return attr