| | |
|---|
| | | |
|---|
| | | using namespace std; |
|---|
| | | |
|---|
| | | extern float *loadparams(const char *filename); |
|---|
| | | extern float *LoadParams(const char *filename); |
|---|
| | | |
|---|
| | | extern void SaveDataFile(const char *filename, void *data, uint32_t len); |
|---|
| | | extern void relu(Tensor<float> *din); |
|---|
| | | extern void swish(Tensor<float> *din); |
|---|
| | | extern void sigmoid(Tensor<float> *din); |
|---|
| | | extern void doubleswish(Tensor<float> *din); |
|---|
| | | extern void Relu(Tensor<float> *din); |
|---|
| | | extern void Swish(Tensor<float> *din); |
|---|
| | | extern void Sigmoid(Tensor<float> *din); |
|---|
| | | extern void DoubleSwish(Tensor<float> *din); |
|---|
| | | |
|---|
| | | extern void softmax(float *din, int mask, int len); |
|---|
| | | extern void Softmax(float *din, int mask, int len); |
|---|
| | | |
|---|
| | | extern void log_softmax(float *din, int len); |
|---|
| | | extern int val_align(int val, int align); |
|---|
| | | extern void disp_params(float *din, int size); |
|---|
| | | extern void LogSoftmax(float *din, int len); |
|---|
| | | extern int ValAlign(int val, int align); |
|---|
| | | extern void DispParams(float *din, int size); |
|---|
| | | |
|---|
| | | extern void basic_norm(Tensor<float> *&din, float norm); |
|---|
| | | extern void BasicNorm(Tensor<float> *&din, float norm); |
|---|
| | | |
|---|
| | | extern void findmax(float *din, int len, float &max_val, int &max_idx); |
|---|
| | | extern void FindMax(float *din, int len, float &max_val, int &max_idx); |
|---|
| | | |
|---|
| | | extern void glu(Tensor<float> *din, Tensor<float> *dout); |
|---|
| | | extern void Glu(Tensor<float> *din, Tensor<float> *dout); |
|---|
| | | |
|---|
| | | string pathAppend(const string &p1, const string &p2); |
|---|
| | | string PathAppend(const string &p1, const string &p2); |
|---|
| | | |
|---|
| | | #endif |
|---|
