Searched refs:output_tile (Results 1 – 5 of 5) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClWinogradConv2d.cpp | 53 Size2D output_tile = Size2D{}; in winograd_output_tile()
64 output_tile = is_input_lt4_nchw ? Size2D(2U, 2U) : Size2D(4U, 4U); in winograd_output_tile()
68 output_tile = is_input_lt4_nchw ? Size2D(2U, 1U) : Size2D(4U, 1U); in winograd_output_tile()
72 output_tile = is_input_lt4_nchw ? Size2D(1U, 2U) : Size2D(1U, 4U); in winograd_output_tile()
77 output_tile = Size2D(kernel_dims.width == 1 ? 1U : 4U, in winograd_output_tile()
82 output_tile = Size2D(kernel_dims.width == 1 ? 1U : 2U, in winograd_output_tile()
86 return output_tile; in winograd_output_tile()
89 bool check_support_fast_math(const Size2D &output_tile, const Size2D &kernel_size) in check_support_fast_math() argument
100 auto p = std::make_pair(std::pair<int, int>(output_tile.width, output_tile.height), in check_support_fast_math()
116 const Size2D output_tile = winograd_output_tile(input_dims, kernel_size, src->data_layout()); in validate_arguments() local
[all …]
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | WinogradConvolutionLayerFixture.h | 191 Size2D output_tile(4U, 4U); in compute_reference()
194 output_tile.width = 2; in compute_reference()
195 output_tile.height = 1; in compute_reference()
199 output_tile.width = 1; in compute_reference()
200 output_tile.height = 2; in compute_reference()
204 output_tile.width = 1; in compute_reference()
208 output_tile.height = 1; in compute_reference()
211 WinogradInfo winograd_info(output_tile, in compute_reference()
359 …void setup(TensorShape input_shape, Size2D output_tile, DataLayout data_layout, DataType data_type) in setup() argument
361 …WinogradInfo winograd_info(output_tile, Size2D(input_shape[0], input_shape[1]), Size2D() /* Not ne… in setup()
|
| /external/ComputeLibrary/src/core/CL/ |
| D | CLHelpers.cpp | 290 bool cl_winograd_convolution_layer_supported(const Size2D &output_tile, const Size2D &kernel_size, … in cl_winograd_convolution_layer_supported() argument
323 auto p = std::make_pair(std::pair<int, int>(output_tile.width, output_tile.height), in cl_winograd_convolution_layer_supported()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Winograd.cpp | 487 SimpleTensor<T> output_tile{ out_tile_dims, in.data_type(), 1 }; in winograd_output_transform() local
526 const int step_y_transf_tile = kernel_size.width == 1 ? 1 : output_tile.shape()[0]; in winograd_output_transform()
547 matrix_multiply<T>(tmp_tile, trans_matrix_transposed, output_tile); in winograd_output_transform()
563 … out[output_offset + yi * stridey_out + xi] = output_tile[xi + yi * step_y_transf_tile]; in winograd_output_transform()
|
| /external/ComputeLibrary/arm_compute/core/CL/ |
| D | CLHelpers.h | 182 bool cl_winograd_convolution_layer_supported(const Size2D &output_tile, const Size2D &kernel_size, …
|