PTO-TILE-MODEL-STATE-FEATURE-MAP-DESCRIPTORS

1// PTO-UNIT: {"id":"PTO-TILE-MODEL-STATE-FEATURE-MAP-DESCRIPTORS","surface":"tile","classification":["model","state","feature-map-descriptors"],"depends_on":["PTO-TILE-MODEL-STATE-LOCAL-REGISTERS"]}2// PTO-STATE: {"id":"PTO-STATE-TILE-FEATURE-MAP","classification":["tile","feature-map"],"scope":"core","owner":"PTO-TILE-MODEL-STATE-FEATURE-MAP-DESCRIPTORS","members":["_TileFeatureMapDescriptors"],"depends_on":["PTO-STATE-TILE-LOCAL"]}3
4type TileFeatureMapLayout of enumeration {5    TileFeatureMapLayout_NC1HWC0,6    TileFeatureMapLayout_NDC1HWC07};8
9type TileFeatureMapDescriptor of record {10    valid: boolean,11    layout: TileFeatureMapLayout,12    batches: integer {1..65535},13    depth: integer {1..65535},14    channel_groups: integer {1..65535},15    height: integer {1..65535},16    width: integer {1..65535},17    channels_per_group: integer {1..65535},18    filter_height: integer {1..65535},19    filter_width: integer {1..65535},20    stride_height: integer {1..65535},21    stride_width: integer {1..65535},22    dilation_height: integer {1..65535},23    dilation_width: integer {1..65535},24    pad_left: integer {0..65535},25    pad_right: integer {0..65535},26    pad_top: integer {0..65535},27    pad_bottom: integer {0..65535},28    logical_channels: integer {1..65535},29    padding: Word,30    transposed: boolean31};32
33var _TileFeatureMapDescriptors :34    array [[PTO_TILE_REGISTER_COUNT]] of TileFeatureMapDescriptor;35
36readonly func ReadTileFeatureMapDescriptor(37    index: TileIndex) => TileFeatureMapDescriptor38begin39    return _TileFeatureMapDescriptors[[index]];40end;41
42func InvalidateTileFeatureMapDescriptor(index: TileIndex)43begin44    _TileFeatureMapDescriptors[[index]].valid = FALSE;45end;46
47func ConfigureTileFeatureMapDescriptor(48    index: TileIndex,49    layout: TileFeatureMapLayout,50    batches: integer {1..65535},51    depth: integer {1..65535},52    channel_groups: integer {1..65535},53    height: integer {1..65535},54    width: integer {1..65535},55    channels_per_group: integer {1..65535},56    filter_height: integer {1..65535},57    filter_width: integer {1..65535},58    stride_height: integer {1..65535},59    stride_width: integer {1..65535},60    dilation_height: integer {1..65535},61    dilation_width: integer {1..65535},62    pad_left: integer {0..65535},63    pad_right: integer {0..65535},64    pad_top: integer {0..65535},65    pad_bottom: integer {0..65535},66    logical_channels: integer {1..65535},67    padding: Word,68    transposed: boolean)69begin70    assert _Tiles[[index]].allocated;71    _TileFeatureMapDescriptors[[index]].valid = TRUE;72    _TileFeatureMapDescriptors[[index]].layout = layout;73    _TileFeatureMapDescriptors[[index]].batches = batches;74    _TileFeatureMapDescriptors[[index]].depth = depth;75    _TileFeatureMapDescriptors[[index]].channel_groups = channel_groups;76    _TileFeatureMapDescriptors[[index]].height = height;77    _TileFeatureMapDescriptors[[index]].width = width;78    _TileFeatureMapDescriptors[[index]].channels_per_group =79        channels_per_group;80    _TileFeatureMapDescriptors[[index]].filter_height = filter_height;81    _TileFeatureMapDescriptors[[index]].filter_width = filter_width;82    _TileFeatureMapDescriptors[[index]].stride_height = stride_height;83    _TileFeatureMapDescriptors[[index]].stride_width = stride_width;84    _TileFeatureMapDescriptors[[index]].dilation_height = dilation_height;85    _TileFeatureMapDescriptors[[index]].dilation_width = dilation_width;86    _TileFeatureMapDescriptors[[index]].pad_left = pad_left;87    _TileFeatureMapDescriptors[[index]].pad_right = pad_right;88    _TileFeatureMapDescriptors[[index]].pad_top = pad_top;89    _TileFeatureMapDescriptors[[index]].pad_bottom = pad_bottom;90    _TileFeatureMapDescriptors[[index]].logical_channels = logical_channels;91    _TileFeatureMapDescriptors[[index]].padding = padding;92    _TileFeatureMapDescriptors[[index]].transposed = transposed;93end;94
95readonly func TileFeatureMapDescriptorStructurallyValid(96    index: TileIndex) => boolean97begin98    let descriptor = ReadTileFeatureMapDescriptor(index);99    if !descriptor.valid || descriptor.transposed then100        return FALSE;101    end;102    if descriptor.layout == TileFeatureMapLayout_NC1HWC0 &&103       descriptor.depth != 1 then104        return FALSE;105    end;106    if descriptor.logical_channels >107       descriptor.channel_groups * descriptor.channels_per_group then108        return FALSE;109    end;110    let physical_elements: integer = descriptor.batches * descriptor.depth *111        descriptor.channel_groups * descriptor.height * descriptor.width *112        descriptor.channels_per_group;113    return physical_elements <=114               TileLogicalElementCapacity(_Tiles[[index]].capacity_bytes,115                   _Tiles[[index]].data_type) &&116           physical_elements <=117               _Tiles[[index]].rows * _Tiles[[index]].columns;118end;119