Searched refs:conv_w (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFusionFixture.h | 94 …TensorType conv_w = create_tensor<TensorType>(w_shape, _data_type, 1, QuantizationInfo(), _data_… in compute_target() local
110 …fuse_fn.configure(&conv_w, &bn_mean, &bn_var, &fused_w, &fused_b, conv_b_ptr, beta_ptr, gamma_ptr,… in compute_target()
114 ARM_COMPUTE_ASSERT(conv_w.info()->is_resizable()); in compute_target()
126 conv_w.allocator()->allocate(); in compute_target()
137 ARM_COMPUTE_ASSERT(!conv_w.info()->is_resizable()); in compute_target()
149 AccessorType(conv_w), AccessorType(conv_b), in compute_target()
163 SimpleTensor<T> conv_w{ w_shape, _data_type, 1 }; in compute_reference()
171 fill(src, conv_w, conv_b, bn_mean, bn_var, bn_beta, bn_gamma); in compute_reference()
174 auto conv_res = reference::convolution_layer(src, conv_w, conv_b, dst_shape, info); in compute_reference()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmConv2d.cpp | 63 unsigned int conv_w = 0; in skip_im_col_info() local
65 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in skip_im_col_info()
291 unsigned int conv_w = 0; in configure() local
293 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in configure()
300 …ARM_COMPUTE_ERROR_ON_MSG((dst->dimension(idx_width) != conv_w) || (dst->dimension(idx_height) != c… in configure()
341 shape_gemm.set(1, conv_w * conv_h); in configure()
370 _col2im_kernel->configure(gemm_output_to_use, dst, Size2D(conv_w, conv_h)); in configure()
400 unsigned int conv_w = 0; in has_opt_impl() local
402 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in has_opt_impl()
461 unsigned int conv_w = 0; in validate() local
[all …]
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmConv2d.cpp | 203 unsigned int conv_w = 0; in configure() local
205 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in configure()
257 shape_gemm.set(1, conv_w * conv_h); in configure()
330 …_col2im_kernel->configure(compile_context, gemm_output_to_use, dst, Size2D(conv_w, conv_h), conv2d… in configure()
334 …ARM_COMPUTE_ERROR_ON_MSG((dst->dimension(idx_width) != conv_w) || (dst->dimension(idx_height) != c… in configure()
417 unsigned int conv_w = 0; in validate() local
420 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in validate()
467 shape_gemm.set(1, conv_w * conv_h); in validate()
529 …_ERROR(kernels::ClCol2ImKernel::validate(gemm_output_to_use, dst, Size2D(conv_w, conv_h), conv2d_i… in validate()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 88 inline TensorShape compute_vector_to_tensor_output_shape(const TensorShape &input, size_t conv_w, s… in compute_vector_to_tensor_output_shape() argument
95 output_shape.set(idx_w, conv_w); in compute_vector_to_tensor_output_shape()
97 output_shape.set(idx_c, input.x() / (conv_w * conv_h)); in compute_vector_to_tensor_output_shape()
|