Go to the source code of this file.
|
template<typename T , typename std::enable_if< is_floating_point< T >::value, int >::type > |
SimpleTensor< T > | yolo_layer (const SimpleTensor< T > &src, const ActivationLayerInfo &info, int32_t num_classes) |
|
template<> |
SimpleTensor< uint8_t > | yolo_layer< uint8_t > (const SimpleTensor< uint8_t > &src, const ActivationLayerInfo &info, int32_t num_classes) |
|
template SimpleTensor< float > | yolo_layer (const SimpleTensor< float > &src, const ActivationLayerInfo &info, int32_t num_classes) |
|
template SimpleTensor< half > | yolo_layer (const SimpleTensor< half > &src, const ActivationLayerInfo &info, int32_t num_classes) |
|