Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
linear.h
Go to the documentation of this file.
1
#ifndef __NN_Linear_H
2
#define __NN_Linear_H
3
4
#include <assert.h>
5
6
#include "
tensor.h
"
7
#include "
matmul.h
"
8
#include "
add.h
"
9
10
21
void
NN_linear
(
Tensor
*y,
const
Tensor
*x,
const
Tensor
*w,
const
Tensor
*b);
22
23
#endif
// __NN_Linear_H
NN_linear
void NN_linear(Tensor *y, const Tensor *x, const Tensor *w, const Tensor *b)
matmul.h
add.h
Tensor
Definition:
tensor.h:36
tensor.h
nn
functional
linear.h
Generated by
1.9.5