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