PTO-TILE-MODEL-LEGALITY-IMAGE-TO-COLUMN

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-LEGALITY-IMAGE-TO-COLUMN","surface":"tile","classification":["model","legality","image-to-column"],"depends_on":["PTO-TILE-MODEL-LEGALITY-DTYPE-LAYOUT","PTO-TILE-MODEL-STATE-FEATURE-MAP-DESCRIPTORS"]}2
3readonly func TileImageToColumnOutputHeight(4    descriptor: TileFeatureMapDescriptor) => integer5begin6    let receptive_height: integer =7        descriptor.dilation_height * (descriptor.filter_height - 1) + 1;8    if descriptor.height + descriptor.pad_top + descriptor.pad_bottom <9       receptive_height then10        return 0;11    end;12    let padded_height: integer =13        descriptor.height + descriptor.pad_top + descriptor.pad_bottom;14    return ((padded_height - receptive_height) DIVRM15            descriptor.stride_height) + 1;16end;17
18readonly func TileImageToColumnOutputWidth(19    descriptor: TileFeatureMapDescriptor) => integer20begin21    let receptive_width: integer =22        descriptor.dilation_width * (descriptor.filter_width - 1) + 1;23    if descriptor.width + descriptor.pad_left + descriptor.pad_right <24       receptive_width then25        return 0;26    end;27    let padded_width: integer =28        descriptor.width + descriptor.pad_left + descriptor.pad_right;29    return ((padded_width - receptive_width) DIVRM30            descriptor.stride_width) + 1;31end;32
33readonly func TileImageToColumnSourceIndex(34    descriptor: TileFeatureMapDescriptor,35    batch: integer,36    depth: integer,37    channel_group: integer,38    input_row: integer,39    input_column: integer,40    channel_in_group: integer) => integer41begin42    return (((((batch * descriptor.depth + depth) *43                descriptor.channel_groups + channel_group) *44               descriptor.height + input_row) *45              descriptor.width + input_column) *46             descriptor.channels_per_group) + channel_in_group;47end;48
49readonly func TileImageToColumnReferencedSourcesDefined(50    source: TileIndex,51    destination: TileIndex,52    position_m: integer {0..65535},53    position_k: integer {0..65535},54    output_height: integer,55    output_width: integer) => boolean56begin57    let descriptor = ReadTileFeatureMapDescriptor(source);58    let source_tile = _Tiles[[source]];59    let destination_tile = _Tiles[[destination]];60    let rows_per_batch: integer =61        descriptor.depth * output_height * output_width;62    let rows_per_depth: integer = output_height * output_width;63    let columns_per_group: integer = descriptor.filter_height *64        descriptor.filter_width * descriptor.channels_per_group;65    let columns_per_filter_row: integer = descriptor.filter_width *66        descriptor.channels_per_group;67
68    for row = 0 to destination_tile.valid_rows - 1 looplimit 65536 do69        let logical_row = position_m + row;70        let batch = logical_row DIVRM rows_per_batch;71        let batch_remainder = logical_row MOD rows_per_batch;72        let depth = batch_remainder DIVRM rows_per_depth;73        let depth_remainder = batch_remainder MOD rows_per_depth;74        let output_row = depth_remainder DIVRM output_width;75        let output_column = depth_remainder MOD output_width;76        for column = 0 to destination_tile.valid_columns - 177            looplimit 65536 do78            let logical_column = position_k + column;79            let channel_group = logical_column DIVRM columns_per_group;80            let group_remainder = logical_column MOD columns_per_group;81            let filter_row = group_remainder DIVRM columns_per_filter_row;82            let filter_remainder = group_remainder MOD columns_per_filter_row;83            let filter_column =84                filter_remainder DIVRM descriptor.channels_per_group;85            let channel_in_group =86                filter_remainder MOD descriptor.channels_per_group;87            let logical_channel: integer = channel_group *88                descriptor.channels_per_group + channel_in_group;89            let padded_input_row: integer =90                output_row * descriptor.stride_height +91                filter_row * descriptor.dilation_height;92            let input_row = padded_input_row - descriptor.pad_top;93            let padded_input_column: integer =94                output_column * descriptor.stride_width +95                filter_column * descriptor.dilation_width;96            let input_column = padded_input_column - descriptor.pad_left;97            if logical_channel < descriptor.logical_channels &&98               input_row >= 0 && input_row < descriptor.height &&99               input_column >= 0 && input_column < descriptor.width then100                let source_index = TileImageToColumnSourceIndex(101                    descriptor,102                    batch,103                    depth,104                    channel_group,105                    input_row,106                    input_column,107                    channel_in_group);108                if source_index >= TileLogicalElementCapacity(109                       source_tile.capacity_bytes, source_tile.data_type) ||110                   !TileLogicalElementDefined(source_tile,111                       source_index as PackedTileElementIndex) then112                    return FALSE;113                end;114            end;115        end;116    end;117    return TRUE;118end;119
120readonly func TileOperandsLegal_TIMG2COL(121    destination: TileIndex,122    source: TileIndex,123    position_m: integer {0..65535},124    position_k: integer {0..65535}) => boolean125begin126    if !TileDescriptorLegal(destination) ||127       !TileDescriptorLegal(source) ||128       !TileFeatureMapDescriptorStructurallyValid(source) then129        return FALSE;130    end;131    let destination_tile = _Tiles[[destination]];132    let source_tile = _Tiles[[source]];133    let descriptor = ReadTileFeatureMapDescriptor(source);134    if !TileCarrierOnlyDataTypeSupported(source_tile.data_type) ||135       destination_tile.data_type != source_tile.data_type ||136       destination_tile.layout != TileLayout_RowMajor ||137       source_tile.layout != TileLayout_RowMajor ||138       destination_tile.location != TileLocation_Matrix ||139       source_tile.location != TileLocation_Matrix then140        return FALSE;141    end;142    let output_height = TileImageToColumnOutputHeight(descriptor);143    let output_width = TileImageToColumnOutputWidth(descriptor);144    if output_height <= 0 || output_width <= 0 then145        return FALSE;146    end;147    let logical_rows: integer = descriptor.batches * descriptor.depth *148        output_height * output_width;149    let logical_columns: integer = descriptor.channel_groups *150        descriptor.filter_height * descriptor.filter_width *151        descriptor.channels_per_group;152    if position_m + destination_tile.valid_rows > logical_rows ||153       position_k + destination_tile.valid_columns > logical_columns then154        return FALSE;155    end;156    return TileImageToColumnReferencedSourcesDefined(157        source,158        destination,159        position_m,160        position_k,161        output_height,162        output_width);163end;164