Home
last modified time | relevance | path

Searched refs:winograd_info (Results 1 – 11 of 11) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/kernels/
DClWinogradOutputTransformKernel.cpp56 …nst ITensorInfo *bias, const ITensorInfo *output, const WinogradInfo &winograd_info, const Activat… in validate_arguments() argument
62 ARM_COMPUTE_RETURN_ERROR_ON(output->data_layout() != winograd_info.output_data_layout); in validate_arguments()
64 const PadStrideInfo conv_info = winograd_info.convolution_info; in validate_arguments()
65 const Size2D output_tile_size = winograd_info.output_tile_size; in validate_arguments()
66 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments()
67 const Size2D input_dimensions = winograd_info.input_dimensions; in validate_arguments()
68 …nnels = (winograd_info.kernel_size.width + winograd_info.output_tile_size.width - 1) * (winogr… in validate_arguments()
70 …l_winograd_convolution_layer_supported(output_tile_size, kernel_size, winograd_info.output_data_la… in validate_arguments()
91 …= input->clone()->set_tensor_shape(compute_winograd_output_transform_shape(*input, winograd_info)); in validate_arguments()
130 …_context, ITensorInfo *src, ITensorInfo *bias, ITensorInfo *dst, const WinogradInfo &winograd_info, in configure() argument
[all …]
DClWinogradInputTransformKernel.cpp50 …e_arguments(const ITensorInfo *input, const ITensorInfo *output, const WinogradInfo &winograd_info) in validate_arguments() argument
55 const PadStrideInfo conv_info = winograd_info.convolution_info; in validate_arguments()
56 const Size2D output_tile_size = winograd_info.output_tile_size; in validate_arguments()
57 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments()
68 …tput_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*input, winograd_info); in validate_arguments()
77 …te_and_configure_window(ITensorInfo *input, ITensorInfo *output, const WinogradInfo &winograd_info) in validate_and_configure_window() argument
87 const PadStrideInfo conv_info = winograd_info.convolution_info; in validate_and_configure_window()
88 const Size2D output_tile_size = winograd_info.output_tile_size; in validate_and_configure_window()
89 const Size2D kernel_size = winograd_info.kernel_size; in validate_and_configure_window()
113 …ileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const WinogradInfo &winograd_info) in configure() argument
[all …]
DClWinogradFilterTransformKernel.cpp52 …e_arguments(const ITensorInfo *input, const ITensorInfo *output, const WinogradInfo &winograd_info) in validate_arguments() argument
57 const Size2D kernel_size = winograd_info.kernel_size; in validate_arguments()
58 const Size2D output_tile_size = winograd_info.output_tile_size; in validate_arguments()
70 …= input->clone()->set_tensor_shape(compute_winograd_filter_transform_shape(*input, winograd_info)); in validate_arguments()
99 …ileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const WinogradInfo &winograd_info) in configure() argument
104 …dst, src->clone()->set_tensor_shape(compute_winograd_filter_transform_shape(*src, winograd_info))); in configure()
106 ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(src, dst, winograd_info)); in configure()
122 …build_opts.add_option_if(winograd_info.kernel_size.height == 1, "-DWINOGRAD_FILTER_TRANSFORM_HORIZ… in configure()
123 …build_opts.add_option_if(winograd_info.kernel_size.width == 1, "-DWINOGRAD_FILTER_TRANSFORM_VERTIC… in configure()
124 const Size2D kernel_size = winograd_info.kernel_size; in configure()
[all …]
DClWinogradFilterTransformKernel.h62 …leContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const WinogradInfo &winograd_info);
69 …Status validate(const ITensorInfo *src, const ITensorInfo *dst, const WinogradInfo &winograd_info);
DClWinogradInputTransformKernel.h62 …leContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const WinogradInfo &winograd_info);
69 …Status validate(const ITensorInfo *src, const ITensorInfo *dst, const WinogradInfo &winograd_info);
DClWinogradOutputTransformKernel.h64 …_context, ITensorInfo *src, ITensorInfo *bias, ITensorInfo *dst, const WinogradInfo &winograd_info,
72 … const ITensorInfo *bias, const ITensorInfo *dst, const WinogradInfo &winograd_info, const Activat…
/external/ComputeLibrary/tests/validation/fixtures/
DWinogradConvolutionLayerFixture.h211 WinogradInfo winograd_info(output_tile, in compute_reference()
218 …pe = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in compute_reference()
219 … = compute_winograd_filter_transform_shape(TensorInfo(weights_shape, 1, data_type), winograd_info); in compute_reference()
222 …mpute_winograd_output_transform_shape(TensorInfo(batched_gemm_shape, 1, data_type), winograd_info); in compute_reference()
228 …nsform_out = reference::winograd_input_transform<T1>(src_t1, input_transform_shape, winograd_info); in compute_reference()
230 …_out = reference::winograd_filter_transform<T1>(weights_t1, filter_transform_shape, winograd_info); in compute_reference()
232 …rence::winograd_output_transform<T1>(batched_gemm, bias_t1, output_transform_shape, winograd_info); in compute_reference()
247 …void setup(TensorShape input_shape, WinogradInfo winograd_info, DataLayout data_layout, DataType d… in setup() argument
249 …ape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_type), winograd_info); in setup()
251 …_target = compute_target(input_shape, output_shape, winograd_info, data_layout, d… in setup()
[all …]
/external/ComputeLibrary/tests/validation/reference/
DWinograd.cpp236 …form(const SimpleTensor<T> &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) in winograd_input_transform() argument
240 const PadStrideInfo conv_info = winograd_info.convolution_info; in winograd_input_transform()
241 const Size2D output_tile_size = winograd_info.output_tile_size; in winograd_input_transform()
242 const Size2D kernel_size = winograd_info.kernel_size; in winograd_input_transform()
360 …form(const SimpleTensor<T> &in, const TensorShape &output_shape, const WinogradInfo &winograd_info) in winograd_filter_transform() argument
367 const Size2D output_tile_size = winograd_info.output_tile_size; in winograd_filter_transform()
368 const Size2D kernel_size = winograd_info.kernel_size; in winograd_filter_transform()
440 … &in, const SimpleTensor<T> &b, const TensorShape &output_shape, const WinogradInfo &winograd_info) in winograd_output_transform() argument
442 const PadStrideInfo conv_info = winograd_info.convolution_info; in winograd_output_transform()
443 const Size2D input_dimensions = winograd_info.input_dimensions; in winograd_output_transform()
[all …]
DWinograd.h48 …orm(const SimpleTensor<T> &in, const TensorShape &output_shape, const WinogradInfo &winograd_info);
51 …orm(const SimpleTensor<T> &in, const TensorShape &output_shape, const WinogradInfo &winograd_info);
54 …&in, const SimpleTensor<T> &b, const TensorShape &output_shape, const WinogradInfo &winograd_info);
/external/ComputeLibrary/src/gpu/cl/operators/
DClWinogradConv2d.cpp128 const WinogradInfo winograd_info = WinogradInfo(output_tile, in validate_arguments() local
135 …input0_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*src, winograd_info); in validate_arguments()
137 …TE_RETURN_ON_ERROR(kernels::ClWinogradInputTransformKernel::validate(src, &input0, winograd_info)); in validate_arguments()
140 …1_shape = misc::shape_calculator::compute_winograd_filter_transform_shape(*weights, winograd_info); in validate_arguments()
142 …TURN_ON_ERROR(kernels::ClWinogradFilterTransformKernel::validate(weights, &input1, winograd_info)); in validate_arguments()
152 …WinogradOutputTransformKernel::validate(&batched_mm_output, biases, dst, winograd_info, act_info)); in validate_arguments()
195 const WinogradInfo winograd_info = WinogradInfo(output_tile, in configure() local
204 _input_transform->configure(compile_context, src, &_input0, winograd_info); in configure()
208 _filter_transform->configure(compile_context, weights, &_input1, winograd_info); in configure()
218 …_output_transform->configure(compile_context, &_batched_mm_output, biases, dst, winograd_info, act… in configure()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h618 …ompute_winograd_filter_transform_shape(const ITensorInfo &input, const WinogradInfo &winograd_info) in compute_winograd_filter_transform_shape() argument
622 const Size2D kernel_size = winograd_info.kernel_size; in compute_winograd_filter_transform_shape()
623 const Size2D output_tile_size = winograd_info.output_tile_size; in compute_winograd_filter_transform_shape()
641 …compute_winograd_input_transform_shape(const ITensorInfo &input, const WinogradInfo &winograd_info) in compute_winograd_input_transform_shape() argument
643 const PadStrideInfo conv_info = winograd_info.convolution_info; in compute_winograd_input_transform_shape()
644 const Size2D kernel_size = winograd_info.kernel_size; in compute_winograd_input_transform_shape()
645 const Size2D output_tile_size = winograd_info.output_tile_size; in compute_winograd_input_transform_shape()
677 …ompute_winograd_output_transform_shape(const ITensorInfo &input, const WinogradInfo &winograd_info) in compute_winograd_output_transform_shape() argument
679 const PadStrideInfo conv_info = winograd_info.convolution_info; in compute_winograd_output_transform_shape()
680 const Size2D kernel_size = winograd_info.kernel_size; in compute_winograd_output_transform_shape()
[all …]