Home
last modified time | relevance | path

Searched refs:idx_w (Results 1 – 20 of 20) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLDeconvolutionReshapeOutputKernel.cpp47 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate_arguments() local
53 ARM_COMPUTE_RETURN_ERROR_ON(weights_info->dimension(idx_w) != deconv_info.stride().first); in validate_arguments()
61 …ARM_COMPUTE_RETURN_ERROR_ON(input->dimension(0) != weights_info->dimension(idx_w) * weights_info->… in validate_arguments()
62 ARM_COMPUTE_RETURN_ERROR_ON(input->dimension(1) != input_info->dimension(idx_w)); in validate_arguments()
82 …on_output_dimensions(input_info->dimension(idx_w), input_info->dimension(idx_h), weights_info->dim… in validate_arguments()
96 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension… in validate_and_configure_window() local
100 …on_output_dimensions(input_info->dimension(idx_w), input_info->dimension(idx_h), weights_info->dim… in validate_and_configure_window()
138 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in configure() local
147 const int filter_w = weights_info->dimension(idx_w); in configure()
150 const int img_w = input_info->dimension(idx_w); in configure()
DCLDeconvolutionLayerUpsampleKernel.cpp54 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate() local
58 ARM_COMPUTE_RETURN_ERROR_ON(output->dimension(idx_w) == 0); in validate()
110 const size_t idx_w = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::WIDTH); in run() local
114 …const int out_end_x = _output->info()->dimension(idx_w) - _info.pad_right() + _info.stride().fir… in run()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLDirectDeconvolutionLayer.cpp66 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate() local
70 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_w) != weights->dimension(idx_h)); in validate()
71 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_w) < 1); in validate()
73 …econvolution_output_dimensions(input->dimension(idx_w), input->dimension(idx_h), weights->dimensio… in validate()
97 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(output->dimension(idx_w) != output_shape[idx_w], "Output's width i… in validate()
136 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in configure() local
144 …put_dimensions(input->info()->dimension(idx_w), input->info()->dimension(idx_h), weights->info()->… in configure()
DCLGEMMDeconvolutionLayer.cpp136 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in validate() local
140 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_w) != deconv_info.stride().first); in validate()
167 …TensorShape gemm_output_shape(weights->dimension(idx_w) * weights->dimension(idx_h) * weights->dim… in validate()
168 input->dimension(idx_w), in validate()
189 …econvolution_output_dimensions(input->dimension(idx_w), input->dimension(idx_h), weights->dimensio… in validate()
DCLDepthwiseConvolutionLayer.cpp171 …const size_t idx_w = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::WI… in validate() local
174 …TURN_ERROR_ON(weights->dimension(idx_w) + (weights->dimension(idx_w) - 1) * (dilation.x() - 1) > i… in validate()
DCLDeconvolutionLayer.cpp151 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in get_deconvolution_method() local
156 …if(weights->dimension(idx_w) != deconv_info.stride().first || weights->dimension(idx_h) != deconv_… in get_deconvolution_method()
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeDefaultConfigValhall.cpp79 …const size_t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimens… in configure_G78_f32() local
82 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f32()
142 …const size_t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimens… in configure_G78_f16() local
145 const size_t src_w = src_shape[idx_w]; in configure_G78_f16()
147 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f16()
245 …const size_t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimens… in configure_G77_f16() local
248 const size_t kernel_w = wei_shape[idx_w]; in configure_G77_f16()
DClDWCNativeDefaultConfigBifrost.cpp46 …const size_t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimens… in configure_f32() local
49 const size_t kernel_w = wei_shape[idx_w]; in configure_f32()
117 …const size_t idx_w = get_data_layout_dimension_index(wei->data_layout(), DataLayoutDimens… in configure_f16() local
120 const size_t src_w = src_shape[idx_w]; in configure_f16()
122 const size_t kernel_w = wei_shape[idx_w]; in configure_f16()
DClDWCNativeHeuristicsHelpers.cpp40 …const size_t idx_w = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimensio… in use_cl_image_for_weights() local
42 const size_t kernel_w = weights->tensor_shape()[idx_w]; in use_cl_image_for_weights()
/external/ComputeLibrary/src/cpu/operators/
DCpuConv2d.cpp130 …const size_t idx_w = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::WI… in get_convolution_method() local
157 …ig) == Size2D(input->dimension(idx_w), input->dimension(idx_h)) && std::get<1>(config) == Size2D(w… in get_convolution_method()
208 …<0>(c) == Size2D(input->dimension(idx_w), input->dimension(idx_h)) && std::get<1>(c) == Size2D(wei… in get_convolution_method()
223 if(weights->dimension(idx_w) == 1 && weights->dimension(idx_h) == 1) in get_convolution_method()
DCpuDepthwiseConv2d.cpp50 …const size_t idx_w = get_data_layout_dimension_index(src->data_layout(), DataLayoutDimension::WIDT… in validate_arguments_optimized() local
52 …URN_ERROR_ON(weights->dimension(idx_w) + (weights->dimension(idx_w) - 1) * (info.dilation.x() - 1)… in validate_arguments_optimized()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEReorgLayerKernel.cpp109 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in run() local
127 const unsigned int w = id[idx_w]; in run()
134 map_coords.set(idx_w, w * stride + offset % stride); in run()
DNEStridedSliceKernel.cpp147 const int idx_w = is_shrink_w ? 0 : index; in run_op() local
192 cur_ptr += (start_3 + (id[idx_w] * shrinked_stride_3)) * byte_increment_3; in run_op()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClWinogradInputTransformKernel.cpp126 const size_t idx_w = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::WIDTH); in configure() local
130 …const Size2D num_tiles = compute_winograd_convolution_tiles(Size2D(src->dimension(idx_w), src->dim… in configure()
150 _src_width = src->dimension(idx_w); in configure()
233 …const size_t idx_w = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::WI… in run_op() local
259 slice.set(idx_w, Window::Dimension(0, _num_tiles_x, 1)); in run_op()
DClWinogradFilterTransformKernel.cpp60 …const size_t idx_w = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::WI… in validate_arguments() local
64 …ARM_COMPUTE_RETURN_ERROR_ON(input->dimension(idx_w) != kernel_size.width || input->dimension(idx_h… in validate_arguments()
/external/ComputeLibrary/src/graph/nodes/
DROIAlignLayerNode.cpp75 …const size_t idx_w = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::WIDT… in configure_output() local
80 output_desc.shape.set(idx_w, _pool_info.pooled_width()); in configure_output()
/external/ComputeLibrary/src/gpu/cl/operators/
DClConv2d.cpp192 …const size_t idx_w = get_data_layout_dimension_index(src->data_layout(), DataLayoutDimension::WIDT… in get_convolution_method() local
222 …onfig) == Size2D(src->dimension(idx_w), src->dimension(idx_h)) && std::get<1>(config) == Size2D(we… in get_convolution_method()
275 …const bool is_large_kernel_sz = (weights->dimension(idx_w) >= kernel_sz_direct_conv_thr) && (weig… in get_convolution_method()
284 … const int32_t kernel_sz = weights->dimension(idx_w) * weights->dimension(idx_h); in get_convolution_method()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEDeconvolutionLayer.cpp134 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension:… in validate() local
136 const unsigned int out_x = (input->dimension(idx_w) - 1) * stride_x + 1; in validate()
138 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_w) > out_x); in validate()
140 ARM_COMPUTE_RETURN_ERROR_ON((out_x - weights->dimension(idx_w) + 1) > out_dims.first); in validate()
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPUpsampleKernel.cpp70 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in run() local
74 const int width_scaled = _output->info()->dimension(idx_w); in run()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h90 const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH); in compute_vector_to_tensor_output_shape() local
95 output_shape.set(idx_w, conv_w); in compute_vector_to_tensor_output_shape()
465 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::W… in compute_deconvolution_upsampled_shape() local
469 unsigned int out_x = (input.dimension(idx_w) - 1) * sx + 1; in compute_deconvolution_upsampled_shape()
473 padx = out_dims.first - (out_x - weights.dimension(idx_w) + 1); in compute_deconvolution_upsampled_shape()
479 scale_out_shape.set(idx_w, out_x); in compute_deconvolution_upsampled_shape()
648 …const size_t idx_w = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimension::WID… in compute_winograd_input_transform_shape() local
653 …const Size2D num_tiles = compute_winograd_convolution_tiles(Size2D(input.tensor_shape()[idx_w], in… in compute_winograd_input_transform_shape()
1217 …const size_t idx_w = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH… in compute_prior_box_shape() local
1222 output_shape.set(0, input.dimension(idx_w) * input.dimension(idx_h) * num_priors * 4); in compute_prior_box_shape()