Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
elu.h
Go to the documentation of this file.
1
#ifndef __NN_ELU_H
2
#define __NN_ELU_H
3
4
#include <assert.h>
5
#include <math.h>
6
7
#include "
tensor.h
"
8
9
22
void
NN_elu
(
Tensor
*y,
const
Tensor
*x,
float
alpha);
23
24
void
NN_elu_inplace
(
Tensor
*x,
float
alpha);
25
26
27
#endif
// __NN_ELU_H
NN_elu_inplace
void NN_elu_inplace(Tensor *x, float alpha)
NN_elu
void NN_elu(Tensor *y, const Tensor *x, float alpha)
Tensor
Definition:
tensor.h:36
tensor.h
nn
functional
elu.h
Generated by
1.9.5