PTO-TILE-MODEL-EXECUTION-IMAGE-TO-COLUMN

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-EXECUTION-IMAGE-TO-COLUMN","surface":"tile","classification":["model","execution","image-to-column"],"depends_on":["PTO-TILE-MODEL-LEGALITY-IMAGE-TO-COLUMN"]}2
3func TIMG2COL(4    destination: TileIndex,5    source: TileIndex,6    position_m: integer {0..65535},7    position_k: integer {0..65535})8begin9    assert TileOperandsLegal_TIMG2COL(10        destination,11        source,12        position_m,13        position_k);14    let descriptor = ReadTileFeatureMapDescriptor(source);15    let source_tile = _Tiles[[source]];16    var result = _Tiles[[destination]];17    let output_height = TileImageToColumnOutputHeight(descriptor);18    let output_width = TileImageToColumnOutputWidth(descriptor);19    let rows_per_batch: integer =20        descriptor.depth * output_height * output_width;21    let rows_per_depth: integer = output_height * output_width;22    let columns_per_group: integer = descriptor.filter_height *23        descriptor.filter_width * descriptor.channels_per_group;24    let columns_per_filter_row: integer = descriptor.filter_width *25        descriptor.channels_per_group;26
27    result.contents_defined = FALSE;28    result.defined_elements = Zeros{PTO_MODEL_TILE_ELEMENTS};29    result.defined_valid_elements = 0;30    result.packed_defined_elements = ZeroPackedTileDefinedElements();31    for row = 0 to result.valid_rows - 1 looplimit 65536 do32        let logical_row = position_m + row;33        let batch = logical_row DIVRM rows_per_batch;34        let batch_remainder = logical_row MOD rows_per_batch;35        let depth = batch_remainder DIVRM rows_per_depth;36        let depth_remainder = batch_remainder MOD rows_per_depth;37        let output_row = depth_remainder DIVRM output_width;38        let output_column = depth_remainder MOD output_width;39        for column = 0 to result.valid_columns - 1 looplimit 65536 do40            let logical_column = position_k + column;41            let channel_group = logical_column DIVRM columns_per_group;42            let group_remainder = logical_column MOD columns_per_group;43            let filter_row = group_remainder DIVRM columns_per_filter_row;44            let filter_remainder = group_remainder MOD columns_per_filter_row;45            let filter_column =46                filter_remainder DIVRM descriptor.channels_per_group;47            let channel_in_group =48                filter_remainder MOD descriptor.channels_per_group;49            let logical_channel: integer = channel_group *50                descriptor.channels_per_group + channel_in_group;51            let padded_input_row: integer =52                output_row * descriptor.stride_height +53                filter_row * descriptor.dilation_height;54            let input_row = padded_input_row - descriptor.pad_top;55            let padded_input_column: integer =56                output_column * descriptor.stride_width +57                filter_column * descriptor.dilation_width;58            let input_column = padded_input_column - descriptor.pad_left;59            var value = TileRawElementValue(60                descriptor.padding,61                source_tile.data_type);62            if logical_channel < descriptor.logical_channels &&63               input_row >= 0 && input_row < descriptor.height &&64               input_column >= 0 && input_column < descriptor.width then65                let source_index = TileImageToColumnSourceIndex(66                    descriptor,67                    batch,68                    depth,69                    channel_group,70                    input_row,71                    input_column,72                    channel_in_group);73                value = TileReadLogicalElement(source_tile,74                    source_index as PackedTileElementIndex);75            end;76            let destination_index = TileLogicalLinearIndex(77                result,78                row as integer {0..65535},79                column as integer {0..65535});80            result = TileInfoWithLogicalElement(result, destination_index,81                value);82        end;83    end;84    result.defined_valid_elements =85        (result.valid_rows * result.valid_columns)86            as integer {0..524288};87    result.contents_defined = TRUE;88    result.location = TileLocation_Matrix;89    result = TileWithPadding(result, TilePad_Null);90    _Tiles[[destination]] = result;91end;92