Esempio n. 1
0
 def extract(cls, node: Node):
     Shape.update_node_stat(
         node, {
             'output_type':
             tf_dtype_extractor(node.pb.attr['out_type'].type, np.int32)
         })
     return cls.enabled
Esempio n. 2
0
 def extract(cls, node):
     Shape.update_node_stat(node, {'output_type': np.int64})
     return cls.enabled
Esempio n. 3
0
 def extract(cls, node):
     Shape.update_node_stat(node, {})
     return cls.enabled