Exemplos de SpaceToDepth.update_node_stat em Python

Exemplo n.º 1

 def extract(cls, node):
     # update the attributes of the node
     block_size = node.pb.attr['block_size'].i
     data_format = node.pb.attr['data_format'].s.decode('utf-8')
     SpaceToDepth.update_node_stat(node, {
         'block_size': block_size,
         'data_format': data_format
     })
     return cls.enabled

Exemplo n.º 2

Arquivo: space_to_depth_ext.py Projeto: yury-intel/openvino

 def extract(cls, node):
     # update the attributes of the node
     block_size = onnx_attr(node, 'blocksize', 'i', default=None)
     SpaceToDepth.update_node_stat(node, {'block_size': block_size})
     return cls.enabled