Home
last modified time | relevance | path

Searched refs:idx_h (Results 1 – 19 of 19) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLDeconvolutionReshapeOutputKernel.cpp48 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate_arguments() local
54 ARM_COMPUTE_RETURN_ERROR_ON(weights_info->dimension(idx_h) != deconv_info.stride().second); in validate_arguments()
61 …>dimension(0) != weights_info->dimension(idx_w) * weights_info->dimension(idx_h) * weights_info->d… in validate_arguments()
63 ARM_COMPUTE_RETURN_ERROR_ON(input->dimension(2) != input_info->dimension(idx_h)); in validate_arguments()
82 …->dimension(idx_w), input_info->dimension(idx_h), weights_info->dimension(idx_w), weights_info->di… in validate_arguments()
97 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension… in validate_and_configure_window() local
100 …->dimension(idx_w), input_info->dimension(idx_h), weights_info->dimension(idx_w), weights_info->di… in validate_and_configure_window()
139 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in configure() local
148 const int filter_h = weights_info->dimension(idx_h); in configure()
151 const int img_h = input_info->dimension(idx_h); in configure()
DCLDeconvolutionLayerUpsampleKernel.cpp55 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate() local
59 ARM_COMPUTE_RETURN_ERROR_ON(output->dimension(idx_h) == 0); in validate()
111 const size_t idx_h = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::HEIGHT); in run() local
118 …const int out_end_y = _output->info()->dimension(idx_h) - _info.pad_bottom() + _info.stride().se… in run()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLDirectDeconvolutionLayer.cpp67 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate() local
70 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_w) != weights->dimension(idx_h)); in validate()
73 …ions(input->dimension(idx_w), input->dimension(idx_h), weights->dimension(idx_w), weights->dimensi… in validate()
98 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(output->dimension(idx_h) != output_shape[idx_h], "Output's height … in validate()
137 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in configure() local
144 …nsion(idx_w), input->info()->dimension(idx_h), weights->info()->dimension(idx_w), weights->info()-… in configure()
DCLGEMMDeconvolutionLayer.cpp137 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in validate() local
141 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_h) != deconv_info.stride().second); in validate()
167 …TensorShape gemm_output_shape(weights->dimension(idx_w) * weights->dimension(idx_h) * weights->dim… in validate()
169 input->dimension(idx_h), in validate()
173 GEMMInfo gemm_info(false, false, true, input->dimension(idx_h), true); in validate()
189 …ions(input->dimension(idx_w), input->dimension(idx_h), weights->dimension(idx_w), weights->dimensi… in validate()
267 …const size_t idx_h = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimen… in configure() local
268 GEMMInfo gemm_info(false, false, true, input->info()->dimension(idx_h), true); in configure()
DCLDepthwiseConvolutionLayer.cpp172 …const size_t idx_h = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::HE… in validate() local
175 …TURN_ERROR_ON(weights->dimension(idx_h) + (weights->dimension(idx_h) - 1) * (dilation.y() - 1) > i… in validate()
DCLDeconvolutionLayer.cpp152 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); 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/gpu/cl/operators/
DClConv2d.cpp193 …const size_t idx_h = get_data_layout_dimension_index(src->data_layout(), DataLayoutDimension::HEIG… in get_convolution_method() local
222 …ension(idx_w), src->dimension(idx_h)) && std::get<1>(config) == Size2D(weights->dimension(idx_w), … in get_convolution_method()
242 …if((src->dimension(idx_h) > 720U) && (dst->dimension(idx_h) > 720U) && (weights->dimension(idx_h) … in get_convolution_method()
247 …if((weights->dimension(idx_h) > 5) && (src->dimension(idx_c) > dst->dimension(idx_c)) && (CLFFTCon… in get_convolution_method()
264 …if((src->dimension(idx_h) > 720U) && (dst->dimension(idx_h) > 720U) && (weights->dimension(idx_h) … in get_convolution_method()
275 …ts->dimension(idx_w) >= kernel_sz_direct_conv_thr) && (weights->dimension(idx_h) >= kernel_sz_dire… 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/cpu/operators/
DCpuConv2d.cpp131 …const size_t idx_h = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::HE… in get_convolution_method() local
157 …sion(idx_w), input->dimension(idx_h)) && std::get<1>(config) == Size2D(weights->dimension(idx_w), … in get_convolution_method()
176 if(input->total_size() > 1e7 && (weights->dimension(idx_h) > 7) in get_convolution_method()
181 …if((weights->dimension(idx_h) > 7) && (input->dimension(idx_c) > output->dimension(idx_c)) && (NEF… in get_convolution_method()
208 …mension(idx_w), input->dimension(idx_h)) && std::get<1>(c) == Size2D(weights->dimension(idx_w), we… in get_convolution_method()
223 if(weights->dimension(idx_w) == 1 && weights->dimension(idx_h) == 1) in get_convolution_method()
DCpuDepthwiseConv2d.cpp51 …const size_t idx_h = get_data_layout_dimension_index(src->data_layout(), DataLayoutDimension::HEIG… in validate_arguments_optimized() local
54 …URN_ERROR_ON(weights->dimension(idx_h) + (weights->dimension(idx_h) - 1) * (info.dilation.y() - 1)… in validate_arguments_optimized()
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeHeuristicsHelpers.cpp41 …const size_t idx_h = get_data_layout_dimension_index(weights->data_layout(), DataLayoutDimensio… in use_cl_image_for_weights() local
43 const size_t kernel_h = weights->tensor_shape()[idx_h]; in use_cl_image_for_weights()
/external/ComputeLibrary/src/core/NEON/kernels/
DNEReorgLayerKernel.cpp110 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in run() local
128 const unsigned int h = id[idx_h]; in run()
135 map_coords.set(idx_h, h * stride + offset / stride); in run()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClWinogradInputTransformKernel.cpp127 const size_t idx_h = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::HEIGHT); in configure() local
130 …um_tiles = compute_winograd_convolution_tiles(Size2D(src->dimension(idx_w), src->dimension(idx_h)), in configure()
151 _src_height = src->dimension(idx_h); in configure()
234 …const size_t idx_h = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::HE… in run_op() local
260 slice.set(idx_h, Window::Dimension(0, _num_tiles_y, 1)); in run_op()
DClWinogradFilterTransformKernel.cpp61 …const size_t idx_h = get_data_layout_dimension_index(input->data_layout(), DataLayoutDimension::HE… in validate_arguments() local
64 …ERROR_ON(input->dimension(idx_w) != kernel_size.width || input->dimension(idx_h) != kernel_size.he… in validate_arguments()
/external/ComputeLibrary/src/graph/nodes/
DROIAlignLayerNode.cpp74 …const size_t idx_h = get_data_layout_dimension_index(output_desc.layout, DataLayoutDimension::HEIG… in configure_output() local
79 output_desc.shape.set(idx_h, _pool_info.pooled_height()); in configure_output()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEDeconvolutionLayer.cpp135 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension:… in validate() local
137 const unsigned int out_y = (input->dimension(idx_h) - 1) * stride_y + 1; in validate()
139 ARM_COMPUTE_RETURN_ERROR_ON(weights->dimension(idx_h) > out_y); in validate()
141 ARM_COMPUTE_RETURN_ERROR_ON((out_y - weights->dimension(idx_h) + 1) > out_dims.second); in validate()
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPUpsampleKernel.cpp71 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in run() local
75 const int height_scaled = _output->info()->dimension(idx_h); in run()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h91 const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT); in compute_vector_to_tensor_output_shape() local
96 output_shape.set(idx_h, conv_h); in compute_vector_to_tensor_output_shape()
466 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::H… in compute_deconvolution_upsampled_shape() local
470 unsigned int out_y = (input.dimension(idx_h) - 1) * sy + 1; in compute_deconvolution_upsampled_shape()
474 pady = out_dims.second - (out_y - weights.dimension(idx_h) + 1); in compute_deconvolution_upsampled_shape()
480 scale_out_shape.set(idx_h, out_y); in compute_deconvolution_upsampled_shape()
649 …const size_t idx_h = get_data_layout_dimension_index(input.data_layout(), DataLayoutDimension::HEI… in compute_winograd_input_transform_shape() local
653 …ompute_winograd_convolution_tiles(Size2D(input.tensor_shape()[idx_w], input.tensor_shape()[idx_h]), in compute_winograd_input_transform_shape()
1218 …const size_t idx_h = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGH… 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()
/external/libaom/av1/encoder/
Dtx_search.c1130 int idx_h[4] = { 0, 1, 2, 3 }; in prune_txk_type_separ() local
1185 sort_rd(rds_h, idx_h, 4); in prune_txk_type_separ()
1187 if (rds_h[idx] > rds_h[0] * 1.2) skip_h[idx_h[idx]] = 1; in prune_txk_type_separ()
1190 if (skip_h[idx_h[0]]) return (uint16_t)0xFFFF; in prune_txk_type_separ()
1195 const int *idx_map_v = idx_map + idx_h[0]; in prune_txk_type_separ()
1234 tx_type = idx_map[idx_v[i_v] * 4 + idx_h[i_h]]; in prune_txk_type_separ()
1235 if (!(allowed_tx_mask & (1 << tx_type)) || skip_h[idx_h[i_h]] || in prune_txk_type_separ()
/external/vixl/src/aarch64/
Dsimulator-aarch64.cc3038 int idx_h = instr->ExtractBits(20, 19); in SimulateSVEComplexIntMulAdd() local
3046 cmla(kFormatVnH, zda, zda, zn, zm_h, idx_h, rot); in SimulateSVEComplexIntMulAdd()
3055 sqrdcmlah(kFormatVnH, zda, zda, zn, zm_h, idx_h, rot); in SimulateSVEComplexIntMulAdd()