コード例 #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
コード例 #2
0
ファイル: shape_ext.py プロジェクト: mikhailk62/openvino 
 def extract(cls, node):
     Shape.update_node_stat(node, {'output_type': np.int64})
     return cls.enabled
コード例 #3
0
 def extract(cls, node):
     Shape.update_node_stat(node, {})
     return cls.enabled