Home
last modified time | relevance | path

Searched refs:weights_height (Results 1 – 10 of 10) sorted by relevance

/external/ComputeLibrary/tests/validation/reference/
DDeconvolutionLayer.cpp48 const int weights_height = weights.shape().y(); in deconvolution_layer() local
49 … int weights_upper_dims = weights.shape().total_size() / (weights_width * weights_height); in deconvolution_layer()
62 unsigned int deconv_pad_y = output_shape.y() - (out_y - weights_height + 1); in deconvolution_layer()
108 const int offset = ud * weights_width * weights_height; in deconvolution_layer()
109 for(int y = 0; y < weights_height; ++y) in deconvolution_layer()
113 …weights_flipped[offset + (weights_height - 1 - y) * weights_width + (weights_width - 1 - x)] = wei… in deconvolution_layer()
DConv3D.cpp68 const unsigned int weights_height = weights.shape()[weights_height_dim]; in calculate_conv3d() local
80 for(unsigned int weight_y = 0; weight_y < weights_height; ++weight_y) in calculate_conv3d()
123 const unsigned int weights_height = weights.shape()[weights_height_dim]; in calculate_conv3d() local
150 for(unsigned int weight_y = 0; weight_y < weights_height; ++weight_y) in calculate_conv3d()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClTransposedConvolutionKernel.cpp83 const size_t weights_height = weights->dimension(height_idx); in validate_arguments() local
85 …nvolution_output_dimensions(input_width, input_height, weights_width, weights_height, deconv_info); in validate_arguments()
114 const size_t weights_height = weights->dimension(height_idx); in configure() local
120 …nvolution_output_dimensions(input_width, input_height, weights_width, weights_height, deconv_info); in configure()
127 const uint32_t pad_top = weights_height - deconv_info.pad_top() - 1; in configure()
169 build_options.add_option("-DWEI_HEIGHT=" + support::cpp11::to_string(weights_height)); in configure()
DClDirectConv3dKernel.cpp111 const size_t weights_height = src1->dimension(3); in configure() local
137 build_options.add_option("-DWEI_HEIGHT=" + support::cpp11::to_string(weights_height)); in configure()
202 _config_id += support::cpp11::to_string(weights_height); in configure()
/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
Dimpl.cpp54 const size_t weights_height; member
75 weights_height(weights.dimension(height_idx)), in DepthwiseConvolutionRunInfo()
183 for(uint32_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_multiplier1_fp()
217 for(size_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_multiplier1_fp()
288 for(size_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_generic_fp()
347 …const int32_t k_offset = run_info.weights_width * run_info.weights_height * input_qoffset *… in depthwise_loop_multiplier1_quantized()
390 for(size_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_multiplier1_quantized()
453 for(size_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_multiplier1_quantized()
518 …const int32_t k_offset = run_info.weights_width * run_info.weights_height * input_qoffset *… in depthwise_loop_generic_quantized()
557 for(size_t h = 0; h < run_info.weights_height; ++h) in depthwise_loop_generic_quantized()
[all …]
/external/ComputeLibrary/tests/validate_examples/
Dgraph_depthwiseconvolution.cpp70 weights_height(parser.add_option<SimpleOption<int>>("weights_height", 3)), in DepthConvolutionOptions()
126 weights_height->set_help("Set weights_dimensions height"); in DepthConvolutionOptions()
169 common_params.weights.height = weights_height->value(); in consume_parameters()
228 …SimpleOption<int> *weights_height; /**< weights heigh… member in __anondcaacb1f0111::DepthConvolutionOptions
Dgraph_convolution.cpp70 weights_height(parser.add_option<SimpleOption<int>>("weights_height", 3)), in ConvolutionOptions()
127 weights_height->set_help("Set weights_dimensions height"); in ConvolutionOptions()
169 common_params.weights.height = weights_height->value(); in consume_parameters()
227 SimpleOption<int> *weights_height; /**< weights height */ member in __anon99218fb20111::ConvolutionOptions
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc160 int weights_height = weight_shape_[0]; in CheckShouldSplitDwConv() local
167 weights_height == 5 && weights_width == 5 && in CheckShouldSplitDwConv()
Dconv_2d_builder.cc44 int weights_height, int weights_width, in ComputeSpaceToBatchParams() argument
54 dilation_factors_h_w[0] * (weights_height - 1) + 1; in ComputeSpaceToBatchParams()
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h722 const unsigned int weights_height = weights_shape[idx_height]; in compute_deep_convolution_shape() local
726 …t_height) = scaled_dimensions(input_width, input_height, weights_width, weights_height, conv_info); in compute_deep_convolution_shape()