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
def extract(cls, node): Shape.update_node_stat(node, {'output_type': np.int64}) return cls.enabled
def extract(cls, node): Shape.update_node_stat(node, {}) return cls.enabled