rml_op_depth_to_space_params
Parameters for the RML_OP_DEPTH_TO_SPACE operation that rearranges tensor data from channels into blocks of spatial data.
More specifically, this operation outputs a copy of the input tensor where values from the channel dimension are moved in spatial blocks to the height and width dimensions. For example, rearranges tensor of shape (N, C * bs^2, H, W) to a tensor of shape (N, C, H * bs, W * bs).
API
Graph Manipulation
Items
Name |
Type |
Description |
|---|---|---|
input |
Operation that outputs tensor containing input data. |
|
block_size |
uint32_t |
Size of data blocks for rearrangement. |