Python AttributeProtoの例

コード例 #1

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

コード例 #2

ファイル: util.py プロジェクト: 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

コード例 #3

ファイル: test_attribute.py プロジェクト: lightmatter-ai/onnx2pytorch

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

コード例 #4

ファイル: util.py プロジェクト: semihlab/onnx-connx

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

コード例 #5

ファイル: util.py プロジェクト: 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

コード例 #6

ファイル: util.py プロジェクト: 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

コード例 #7

ファイル: util.py プロジェクト: 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

コード例 #8

ファイル: util.py プロジェクト: semihlab/onnx-connx

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