Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
add.h
Go to the documentation of this file.
1
#ifndef __NN_ADD_H
2
#define __NN_ADD_H
3
4
#include <assert.h>
5
#include <math.h>
6
7
#include "
tensor.h
"
8
#include "
print.h
"
9
#include "
ops/acc.h
"
10
#include "
ops/acc1.h
"
11
#include "
ops/add.h
"
12
#include "
ops/add1.h
"
13
14
26
void
NN_add
(
Tensor
*out,
const
Tensor
*a,
const
Tensor
*b);
27
35
void
NN_add1
(
Tensor
*out,
const
Tensor
*in,
float
scalar);
36
47
void
NN_add_inplace
(
Tensor
*b,
const
Tensor
*a);
48
55
void
NN_add1_inplace
(
Tensor
*b,
float
scalar);
56
57
58
#endif
// __NN_ADD_H
acc1.h
acc.h
add1.h
NN_add1_inplace
void NN_add1_inplace(Tensor *b, float scalar)
NN_add_inplace
void NN_add_inplace(Tensor *b, const Tensor *a)
NN_add1
void NN_add1(Tensor *out, const Tensor *in, float scalar)
NN_add
void NN_add(Tensor *out, const Tensor *a, const Tensor *b)
add.h
print.h
Tensor
Definition:
tensor.h:36
tensor.h
nn
functional
add.h
Generated by
1.9.5