Home
last modified time | relevance | path

Searched refs:conv_h (Results 1 – 3 of 3) sorted by relevance

/external/ComputeLibrary/src/cpu/operators/
DCpuGemmConv2d.cpp64 unsigned int conv_h = 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()
75 …ut::NHWC && (bool(CpuGemmConv2d::validate_gemm3d(src, weights, act_info, conv_h, /*skip_im2col*/ t… in skip_im_col_info()
83 …ut::NHWC && (bool(CpuGemmConv2d::validate_gemm3d(src, weights, act_info, conv_h, /*skip_im2col*/ f… in skip_im_col_info()
292 unsigned int conv_h = 0; in configure() local
293 std::tie(conv_w, conv_h) = scaled_dimensions(src->dimension(idx_width), in configure()
300 …MPUTE_ERROR_ON_MSG((dst->dimension(idx_width) != conv_w) || (dst->dimension(idx_height) != conv_h), in configure()
341 shape_gemm.set(1, conv_w * conv_h); in configure()
362 const unsigned int gemm_3d_depth = _skip_col2im ? conv_h : 0; in configure()
370 _col2im_kernel->configure(gemm_output_to_use, dst, Size2D(conv_w, conv_h)); in configure()
[all …]
/external/ComputeLibrary/src/gpu/cl/operators/
DClGemmConv2d.cpp204 unsigned int conv_h = 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()
319 const unsigned int gemm_3d_depth = (data_layout == DataLayout::NHWC) ? conv_h : 0; in configure()
330 …_col2im_kernel->configure(compile_context, gemm_output_to_use, dst, Size2D(conv_w, conv_h), conv2d… in configure()
334 …MPUTE_ERROR_ON_MSG((dst->dimension(idx_width) != conv_w) || (dst->dimension(idx_height) != conv_h), in configure()
418 unsigned int conv_h = 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()
521 const unsigned int gemm_3d_depth = (data_layout == DataLayout::NHWC) ? conv_h : 0; in validate()
[all …]
/external/ComputeLibrary/arm_compute/core/utils/misc/
DShapeCalculator.h88 …r_to_tensor_output_shape(const TensorShape &input, size_t conv_w, size_t conv_h, const DataLayout … in compute_vector_to_tensor_output_shape() argument
96 output_shape.set(idx_h, conv_h); 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()