Пример #1
0
 def extract(cls, node: Node):
     Shape.update_node_stat(
         node, {
             'data_type':
             tf_dtype_extractor(node.pb.attr['out_type'].type, np.int32)
         })
     return cls.enabled
Пример #2
0
 def extract(cls, node):
     Shape.update_node_stat(node, {})
     return cls.enabled
Пример #3
0
 def extract(cls, node):
     Shape.update_node_stat(node, {'output_type': np.int64})
     return cls.enabled
Пример #4
0
 def extract(node):
     Shape.update_node_stat(node)
     return __class__.enabled