Python Einsum.update_node_statの例

コード例 #1

ファイル: einsum_ext.py プロジェクト: yury-intel/openvino

 def extract(cls, einsum_node):
     einsum_name = einsum_node.soft_get('name', einsum_node.id)
     equation = onnx_attr(einsum_node, 'equation',
                          's').decode(encoding="utf-8")
     normalized_equation = Einsum.normalize_equation(einsum_name, equation)
     Einsum.update_node_stat(einsum_node, {'equation': normalized_equation})
     return cls.enabled

コード例 #2

 def extract(cls, einsum_node):
     einsum_name = einsum_node.soft_get('name', einsum_node.id)
     attrs = get_mxnet_layer_attrs(einsum_node.symbol_dict)
     equation = attrs.str('subscripts')
     normalized_equation = Einsum.normalize_equation(einsum_name, equation)
     Einsum.update_node_stat(einsum_node, {'equation': normalized_equation})
     return cls.enabled

コード例 #3

ファイル: einsum_ext.py プロジェクト: mikhailk62/openvino

 def extract(cls, einsum_node):
     einsum_name = einsum_node.soft_get('name', einsum_node.id)
     equation = einsum_node.pb.attr['equation'].s.decode('utf-8')
     normalized_equation = Einsum.normalize_equation(einsum_name, equation)
     Einsum.update_node_stat(einsum_node, {'equation': normalized_equation})
     return cls.enabled