24 #ifndef ARM_COMPUTE_TEST_YOLO_LAYER_H 25 #define ARM_COMPUTE_TEST_YOLO_LAYER_H 38 template <typename T, typename std::enable_if<is_floating_point<T>::value,
int>
::type = 0>
39 SimpleTensor<T>
yolo_layer(
const SimpleTensor<T> &
src,
const ActivationLayerInfo &
info, int32_t num_classes);
41 template <typename T, typename std::enable_if<std::is_integral<T>::value,
int>::type = 0>
42 SimpleTensor<T>
yolo_layer(
const SimpleTensor<T> &
src,
const ActivationLayerInfo &
info, int32_t num_classes);
decltype(strategy::transforms) typedef type
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
SimpleTensor< T > yolo_layer(const SimpleTensor< T > &src, const ActivationLayerInfo &info, int32_t num_classes)