-
Notifications
You must be signed in to change notification settings - Fork 247
Expand file tree
/
Copy pathuTensor.h
More file actions
47 lines (41 loc) · 1.19 KB
/
uTensor.h
File metadata and controls
47 lines (41 loc) · 1.19 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#ifndef __UTENSOR_H
#define __UTENSOR_H
/*
* Core bits
*/
// This one selects platform specific stuff. probably should come first
#include "uTensor/core/modelBase.hpp"
#include "uTensor/core/operatorBase.hpp"
#include "uTensor/core/uTensor_util.hpp"
/*
* Allocators
*/
#include "uTensor/allocators/arenaAllocator.hpp"
/*
* Operators
*/
#include "uTensor/ops/ActivationFncs.hpp"
#include "uTensor/ops/ArgMinMax.hpp"
#include "uTensor/ops/Arithmetic.hpp"
#include "uTensor/ops/Convolution.hpp"
#include "uTensor/ops/Functional.hpp"
#include "uTensor/ops/Matrix.hpp"
#include "uTensor/ops/Reshape.hpp"
#include "uTensor/ops/Rsqrt.hpp"
#include "uTensor/ops/Sum.hpp"
#include "uTensor/ops/symmetric_quantization/QuantizeOps.hpp"
#include "uTensor/ops/symmetric_quantization/convolution2d.hpp"
#include "uTensor/ops/symmetric_quantization/depthwise_separable_convolution.hpp"
#include "uTensor/ops/symmetric_quantization/fully_connected.hpp"
#include "uTensor/ops/trigonometric.hpp"
/*
* Tensors
*/
#include "uTensor/tensors/BufferTensor.hpp"
#include "uTensor/tensors/RamTensor.hpp"
#include "uTensor/tensors/RomTensor.hpp"
/*
* Error Handlers
*/
#include "uTensor/errorHandlers/SimpleErrorHandler.hpp"
#endif