Compute Library
 21.02
WinogradInfo Struct Reference

Winograd information. More...

#include <Types.h>

Collaboration diagram for WinogradInfo:
[legend]

Public Member Functions

 WinogradInfo (Size2D output_tile_sz, Size2D kernel_sz, Size2D input_dims, PadStrideInfo conv_info, DataLayout data_layout)
 Default constructor. More...
 

Data Fields

Size2D output_tile_size {}
 Width and height of the output tile. More...
 
Size2D kernel_size {}
 Width and height of the kernel. More...
 
Size2D input_dimensions {}
 Width and height of the input tensor before the convolution is applied. More...
 
PadStrideInfo convolution_info {}
 Convolution info (Pads, strides,...) More...
 
DataLayout output_data_layout { DataLayout::NCHW }
 Data layout to use for the output tensor once the convolution has been applied (NCHW or NHWC) More...
 

Detailed Description

Winograd information.

Definition at line 2182 of file Types.h.

Constructor & Destructor Documentation

◆ WinogradInfo()

WinogradInfo ( Size2D  output_tile_sz,
Size2D  kernel_sz,
Size2D  input_dims,
PadStrideInfo  conv_info,
DataLayout  data_layout 
)
inline

Default constructor.

Parameters
[in]output_tile_szWidth and height of the output tile
[in]kernel_szWidth and height of the kernel
[in]input_dimsWidth and height of the input tensor before the convolution is applied
[in]conv_infoConvolution info (Pads, strides)
[in]data_layoutData layout to use for the output tensor once the convolution has been applied

Definition at line 2192 of file Types.h.

2194  {
2195  }
DataLayout output_data_layout
Data layout to use for the output tensor once the convolution has been applied (NCHW or NHWC) ...
Definition: Types.h:2201
PadStrideInfo convolution_info
Convolution info (Pads, strides,...)
Definition: Types.h:2200
const DataLayout data_layout
Definition: Im2Col.cpp:151
Size2D output_tile_size
Width and height of the output tile.
Definition: Types.h:2197
Size2D kernel_size
Width and height of the kernel.
Definition: Types.h:2198
Size2D input_dimensions
Width and height of the input tensor before the convolution is applied.
Definition: Types.h:2199

Field Documentation

◆ convolution_info

◆ input_dimensions

Size2D input_dimensions {}

◆ kernel_size

◆ output_data_layout

DataLayout output_data_layout { DataLayout::NCHW }

◆ output_tile_size


The documentation for this struct was generated from the following file: