Esempi in Python per AttributeProto

Esempio n. 1

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

Esempio n. 2

File: util.py Progetto: 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

Esempio n. 3

File: test_attribute.py Progetto: lightmatter-ai/onnx2pytorch

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

Esempio n. 4

File: util.py Progetto: semihlab/onnx-connx

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

Esempio n. 5

File: util.py Progetto: 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

Esempio n. 6

File: util.py Progetto: 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

Esempio n. 7

File: util.py Progetto: 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

Esempio n. 8

File: util.py Progetto: semihlab/onnx-connx

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