#include "InstanceNormalizationLayer.h"
#include "tests/validation/Helpers.h"
#include <algorithm>
#include <cmath>
Go to the source code of this file.
|
template<typename T > |
SimpleTensor< T > | instance_normalization (const SimpleTensor< T > &src, float gamma, float beta, float epsilon) |
|
template SimpleTensor< float > | instance_normalization (const SimpleTensor< float > &src, float gamma, float beta, float epsilon) |
|
template SimpleTensor< half > | instance_normalization (const SimpleTensor< half > &src, float gamma, float beta, float epsilon) |
|