forked from jiuyuan/InfiniTensor
12 lines
502 B
C
12 lines
502 B
C
|
#pragma once
|
||
|
|
#include "operators/unary.h"
|
||
|
|
|
||
|
|
namespace infini {
|
||
|
|
void LaynormKernel(const float *input, const float *scale, const float eps,
|
||
|
|
int size, int scaleSize, const int dimsize, const int stride,
|
||
|
|
float *output, const float *bias, int biasSize);
|
||
|
|
void LaynormKernel(const float *input, const float *scale, const float eps,
|
||
|
|
int size, int scaleSize, const int dimsize, const int stride,
|
||
|
|
float *output);
|
||
|
|
}; // namespace infini
|