34 #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
36 const Window &window,
const ITensor *
src,
const ITensor *weights, ITensor *
dst,
const PadStrideInfo &
conv_info)
40 #endif //defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
47 std::pair<unsigned int, unsigned int> convolved_dims,
50 uint32_t input_pad_right,
53 #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
54 arm_compute::cpu::kernels::run_im2col<float16_t, true, true>(
56 #else // defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
59 #endif // defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
67 std::pair<unsigned int, unsigned int> convolved_dims,
70 uint32_t input_pad_right,
73 #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
74 arm_compute::cpu::kernels::run_im2col<float16_t, false, true>(
76 #else // defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)
79 #endif // defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS)