21.02
|
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 { DataLayout::NCHW } |
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 2192 of file Types.h.
PadStrideInfo convolution_info {} |
Convolution info (Pads, strides,...)
Definition at line 2200 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::operator<<(), 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 2199 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 2198 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(), CLWinogradInputTransformKernel::configure(), CLWinogradFilterTransformKernel::configure(), CLWinogradOutputTransformKernel::configure(), arm_compute::operator<<(), 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 { DataLayout::NCHW } |
Data layout to use for the output tensor once the convolution has been applied (NCHW or NHWC)
Definition at line 2201 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_winograd_output_transform_shape(), CLWinogradOutputTransformKernel::configure(), arm_compute::operator<<(), and arm_compute::test::validation::reference::winograd_output_transform().
Size2D output_tile_size {} |
Width and height of the output tile.
Definition at line 2197 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(), arm_compute::operator<<(), 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().