#include "L2NormalizeLayer.h"
#include "ReductionOperation.h"
#include "tests/validation/Helpers.h"
#include <algorithm>
#include <cmath>
Go to the source code of this file.
|
template<typename T > |
SimpleTensor< T > | l2_normalize (const SimpleTensor< T > &src, unsigned int axis, float epsilon) |
|
template SimpleTensor< float > | l2_normalize (const SimpleTensor< float > &src, unsigned int axis, float epsilon) |
|
template SimpleTensor< half > | l2_normalize (const SimpleTensor< half > &src, unsigned int axis, float epsilon) |
|