24.02.1
|
Winograd information. More...
#include <Types.h>
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 |
Data layout to use for the output tensor once the convolution has been applied (NCHW or NHWC) More... | |
|
inline |
Default constructor.
[in] | output_tile_sz | Width and height of the output tile |
[in] | kernel_sz | Width and height of the kernel |
[in] | input_dims | Width and height of the input tensor before the convolution is applied |
[in] | conv_info | Convolution info (Pads, strides) |
[in] | data_layout | Data layout to use for the output tensor once the convolution has been applied |
Definition at line 1940 of file Types.h.
PadStrideInfo convolution_info {} |
Convolution info (Pads, strides,...)
Definition at line 1953 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_input_transform_shape(), arm_compute::misc::shape_calculator::compute_winograd_output_transform_shape(), ClWinogradInputTransformKernel::configure(), ClWinogradOutputTransformKernel::configure(), arm_compute::test::validation::reference::winograd_input_transform(), and arm_compute::test::validation::reference::winograd_output_transform().
Size2D input_dimensions {} |
Width and height of the input tensor before the convolution is applied.
Definition at line 1952 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_output_transform_shape(), ClWinogradOutputTransformKernel::configure(), and arm_compute::test::validation::reference::winograd_output_transform().
Size2D kernel_size {} |
Width and height of the kernel.
Definition at line 1951 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_filter_transform_shape(), arm_compute::misc::shape_calculator::compute_winograd_input_transform_shape(), arm_compute::misc::shape_calculator::compute_winograd_output_transform_shape(), ClWinogradFilterTransformKernel::configure(), ClWinogradInputTransformKernel::configure(), ClWinogradOutputTransformKernel::configure(), arm_compute::test::validation::reference::winograd_filter_transform(), arm_compute::test::validation::reference::winograd_input_transform(), and arm_compute::test::validation::reference::winograd_output_transform().
DataLayout output_data_layout |
Data layout to use for the output tensor once the convolution has been applied (NCHW or NHWC)
Definition at line 1954 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_output_transform_shape(), ClWinogradOutputTransformKernel::configure(), and arm_compute::test::validation::reference::winograd_output_transform().
Size2D output_tile_size {} |
Width and height of the output tile.
Definition at line 1950 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_filter_transform_shape(), arm_compute::misc::shape_calculator::compute_winograd_input_transform_shape(), ClWinogradInputTransformKernel::configure(), ClWinogradFilterTransformKernel::configure(), ClWinogradOutputTransformKernel::configure(), ClWinogradOutputTransformKernel::validate(), arm_compute::test::validation::reference::winograd_filter_transform(), arm_compute::test::validation::reference::winograd_input_transform(), and arm_compute::test::validation::reference::winograd_output_transform().