Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
matmul.h
Go to the documentation of this file.
1
#ifndef __NN_MATMUL_H
2
#define __NN_MATMUL_H
3
4
#include <assert.h>
5
6
#include "
tensor.h
"
7
#include "
mm.h
"
8
#include "
mv.h
"
9
10
20
void
NN_matmul
(
Tensor
*out,
const
Tensor
*a,
const
Tensor
*b);
21
31
void
NN_matmul_t
(
Tensor
*out,
const
Tensor
*a,
const
Tensor
*b);
32
33
34
#endif
// __NN_MATMUL_H
NN_matmul_t
void NN_matmul_t(Tensor *out, const Tensor *a, const Tensor *b)
NN_matmul
void NN_matmul(Tensor *out, const Tensor *a, const Tensor *b)
mv.h
mm.h
Tensor
Definition:
tensor.h:36
tensor.h
nn
functional
matmul.h
Generated by
1.9.5