Home
last modified time | relevance | path

Searched refs:height_idx (Results 1 – 23 of 23) sorted by relevance

/external/ComputeLibrary/src/runtime/NEON/functions/
DNEDeconvolutionLayer.cpp90 …const unsigned int height_idx = get_data_layout_dimension_index(weights->data_layout(), DataLayout… in validate() local
91 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(width_idx) != weights->dimension(height_idx)); in validate()
103 …>dimension(width_idx), input->dimension(height_idx), weights->dimension(width_idx), weights->dimen… in validate()
166 …const unsigned int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension:… in configure() local
167 …lution_output_dimensions(input->info()->dimension(width_idx), input->info()->dimension(height_idx), in configure()
168 … weights->info()->dimension(width_idx), weights->info()->dimension(height_idx), info); in configure()
181 … != 1 || weights->info()->dimension(width_idx) != 1 || weights->info()->dimension(height_idx) != 1; in configure()
200 axis_data[1] = static_cast<uint32_t>(height_idx); in configure()
/external/ComputeLibrary/src/cpu/kernels/
DCpuIm2ColKernel.cpp66 …const unsigned int height_idx = get_data_layout_dimension_index(input->data_layout(), DataLayout… in validate_arguments() local
68 …const unsigned total_height = input->dimension(height_idx) + conv_info.pad_top() + conv_info.p… in validate_arguments()
258 …const unsigned int height_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension… in run_im2col() local
262 const int input_h = src->info()->dimension(height_idx); in run_im2col()
286 const int start_h = id[height_idx] * stride_y - pad_top; in run_im2col()
290 … output_ptr = reinterpret_cast<T *>(out.ptr() + (id[width_idx] + id[height_idx] * _convolved_di… in run_im2col()
343 …const unsigned int height_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension… in configure() local
350 _convolved_dims = scaled_dimensions(src->dimension(width_idx), dst->dimension(height_idx), in configure()
413 …gned int> convolved_dims = scaled_dimensions(src->dimension(width_idx), src->dimension(height_idx), in configure()
419 win.set(height_idx, Window::Dimension(0, convolved_dims.second, 1)); in configure()
DCpuWinogradConv2dKernel.cpp44 const unsigned int height_idx = 2; in run_op() local
49 const size_t input_row_stride = src_strides[height_idx] / element_size_in_bytes; in run_op()
83 const unsigned int height_idx = 2; in run_op() local
88 const size_t out_row_stride = dst_strides[height_idx] / element_size_in_bytes; in run_op()
DCpuConvertFullyConnectedWeightsKernel.cpp51 …const int height_idx = get_data_layout_dimension_index(input_data_layout, DataLayoutDimension::HE… in configure() local
54 …int num_elems_per_input_plane = original_input_shape[width_idx] * original_input_shape[height_idx]; in configure()
DCpuDirectConv2dKernel.cpp70 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in validate_arguments() local
74 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(width_idx) != weights->dimension(height_idx)); in validate_arguments()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClTransposedConvolutionKernel.cpp54 constexpr unsigned int height_idx = 2; in validate_arguments() local
81 const size_t input_height = input->dimension(height_idx); in validate_arguments()
83 const size_t weights_height = weights->dimension(height_idx); in validate_arguments()
108 constexpr unsigned int height_idx = 2; in configure() local
112 const size_t input_height = input->dimension(height_idx); in configure()
114 const size_t weights_height = weights->dimension(height_idx); in configure()
116 const size_t output_height = output->dimension(height_idx); in configure()
DClIm2ColKernel.cpp79 …const unsigned int height_idx = get_data_layout_dimension_index(src->data_layout(), DataLayoutDi… in validate_arguments() local
81 …const unsigned total_height = src->dimension(height_idx) + conv_info.pad_top() + conv_info.pad… in validate_arguments()
107 …const unsigned int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension… in validate_and_configure_window() local
109 const unsigned int input_height = src->dimension(height_idx); in validate_and_configure_window()
153 …const unsigned int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimensio… in configure_opencl_kernel() local
156 const unsigned int input_height = src->dimension(height_idx); in configure_opencl_kernel()
315 …const unsigned int height_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimensio… in configure() local
317 const unsigned int input_height = src->dimension(height_idx); in configure()
DClIndirectConv2dAddressPrecalculationKernel.cpp84 constexpr unsigned int height_idx = 2; in configure() local
118 …d_options.add_option("-DSRC_CONV_HEIGHT=" + support::cpp11::to_string(src->dimension(height_idx))); in configure()
120 …ptions.add_option("-DDST_CONV_HEIGHT=" + support::cpp11::to_string(output_conv_shape[height_idx])); in configure()
DClIndirectConv2dKernel.cpp127 constexpr unsigned int height_idx = 2; in configure() local
129 const unsigned int kernel_height = weights->dimension(height_idx); in configure()
205 … build_options.add_option("-DDST_HEIGHT=" + support::cpp11::to_string(dst->dimension(height_idx))); in configure()
243 _config_id += support::cpp11::to_string(src->dimension(height_idx)); in configure()
249 _config_id += support::cpp11::to_string(dst->dimension(height_idx)); in configure()
DClDirectConv2dKernel.cpp61 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in validate_arguments() local
72 …ETURN_ERROR_ON_MSG(weights->dimension(width_idx) != weights->dimension(height_idx), "Weights shoul… in validate_arguments()
167 …const unsigned int height_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension… in configure() local
272 …ld_options.add_option("-DWEI_HEIGHT=" + support::cpp11::to_string(weights->dimension(height_idx))); in configure()
331 … build_options.add_option("-DSRC_HEIGHT=" + support::cpp11::to_string(src->dimension(height_idx))); in configure()
338 …ld_options.add_option("-DWEI_HEIGHT=" + support::cpp11::to_string(weights->dimension(height_idx))); in configure()
390 _config_id += support::cpp11::to_string(dst->dimension(height_idx)); in configure()
DClConvertFullyConnectedWeightsKernel.cpp63 …const int height_idx = get_data_layout_dimension_index(src_data_layout, DataLayoutDimension::HEIG… in configure() local
66 …igned int num_elems_per_src_plane = original_src_shape[width_idx] * original_src_shape[height_idx]; in configure()
/external/ComputeLibrary/src/graph/nodes/
DResizeLayerNode.cpp74 size_t height_idx = get_dimension_idx(data_layout, DataLayoutDimension::HEIGHT); in configure_output() local
76 …output_desc.shape.set(height_idx, static_cast<int>(output_desc.shape[height_idx] * _scale_height)); in configure_output()
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/
DClTemplatePool2d.cpp44 constexpr static int32_t height_idx = 2; variable
146 …().x() == _src->dimension(width_idx) && _attributes.pool_size().y() == _src->dimension(height_idx)) in get_MxN_kernel_code()
411 lut["SRC_HEIGHT"] = _src->dimension(height_idx); in get_tag_lut()
415 lut["DST_HEIGHT"] = _dst->dimension(height_idx); in get_tag_lut()
446 config_id += support::cpp11::to_string(_dst->dimension(height_idx)); in get_config_id()
DClTemplateDepthwiseConv2d.cpp66 constexpr int height_idx = 2; // Data Layout is NHWC in get_component_code() local
116 if(_weight->dimension(height_idx) < 5) in get_component_code()
169 if(_weight->dimension(height_idx) < 5) in get_component_code()
288 constexpr int height_idx = 2; in get_tag_lut() local
291 lut["WEI_HEIGHT"] = _weight->dimension(height_idx); in get_tag_lut()
DClTemplateDirectConv2d.cpp296 const auto height_idx = 2; in get_tag_lut() local
298 lut["WEI_HEIGHT"] = _weight->dimension(height_idx); in get_tag_lut()
347 …const unsigned int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::… in get_config_id() local
362 config_id += support::cpp11::to_string(_dst->dimension(height_idx)); in get_config_id()
/external/ComputeLibrary/tests/validation/reference/
DConvertFullyConnectedWeights.cpp42 …const int height_idx = get_data_layout_dimension_index(original_input_data_layout, DataLayoutDime… in convert_fully_connected_weights() local
46 …int num_elems_per_input_plane = original_input_shape[width_idx] * original_input_shape[height_idx]; in convert_fully_connected_weights()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEPriorBoxLayerKernel.cpp108 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in calculate_prior_boxes() local
111 const int layer_height = _input1->info()->dimension(height_idx); in calculate_prior_boxes()
118 img_height = _input2->info()->dimension(height_idx); in calculate_prior_boxes()
DNESpaceToBatchLayerKernel.cpp162 …const int height_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::HEIGHT… in run() local
167 const size_t height = _input->info()->dimension(height_idx); in run()
/external/ComputeLibrary/src/core/CL/kernels/
DCLPriorBoxLayerKernel.cpp131 … const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure() local
134 const int layer_height = input1->info()->dimension(height_idx); in configure()
141 img_height = input2->info()->dimension(height_idx); in configure()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h380 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… variable
391 col2im_shape.set(height_idx, convolved_dims.height);
428 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in compute_depthwise_convolution_shape() local
437 …e(output_width, output_height) = scaled_dimensions(input_shape[width_idx], input_shape[height_idx], in compute_depthwise_convolution_shape()
443 output_shape.set(height_idx, output_height); in compute_depthwise_convolution_shape()
500 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in compute_deconvolution_output_shape() local
506 out_shape.set(height_idx, out_dims.second); in compute_deconvolution_output_shape()
537 …const int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… variable
540 …nt> out_dims = scaled_dimensions(output_shape[width_idx], output_shape[height_idx], kernel_dims.wi…
/external/ComputeLibrary/src/core/
DUtils.cpp374 …const unsigned int height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimensio… in calculate_same_pad() local
376 const unsigned int in_height = input_shape[height_idx]; in calculate_same_pad()
378 const unsigned int kernel_height = weights_shape[height_idx]; in calculate_same_pad()
/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
Dimpl.cpp36 const size_t height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIG… variable
73 …input_max_offset(input.strides_in_bytes().z() * input.dimension(height_idx) - (input.padding().bot… in DepthwiseConvolutionRunInfo()
75 weights_height(weights.dimension(height_idx)), in DepthwiseConvolutionRunInfo()
82 input_height(input.dimension(height_idx)), in DepthwiseConvolutionRunInfo()
/external/ComputeLibrary/src/cpu/operators/
DCpuWinogradConv2d.cpp386 const unsigned int height_idx = 3; // H in HWIO in prepare() local
390 …ht_row_stride = permuted_weights.get()->info()->strides_in_bytes()[height_idx] / element_size_… in prepare()