Skip to content

Commit fdf2a3e

Browse files
anyj0527myungjoo
authored andcommittedAug 31, 2021
[tensor_sparse] Implement tensor_sparse_enc and tensor_sparse_dec
- tensor_sparse_enc: encoding static tensors into sparse tensors - tensor_sparse_dec: decoding sparse tensors into stsatic tensors Signed-off-by: Yongjoo Ahn <yongjoo1.ahn@samsung.com>
1 parent 3cd87e7 commit fdf2a3e

File tree

8 files changed

+987
-0
lines changed

8 files changed

+987
-0
lines changed
 

‎gst/nnstreamer/meson.build

+1
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ nnst_plugins = [
5555
'tensor_mux',
5656
'tensor_sink',
5757
'tensor_source',
58+
'tensor_sparse',
5859
'tensor_split',
5960
'tensor_transform',
6061
'tensor_filter',

‎gst/nnstreamer/registerer/nnstreamer.c

+4
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,8 @@
6161
#if defined(__gnu_linux__) && !defined(__ANDROID__)
6262
#include <tensor_source/tensor_src_iio.h>
6363
#endif /* __gnu_linux__ && !__ANDROID__ */
64+
#include <tensor_sparse/tensor_sparse_enc.h>
65+
#include <tensor_sparse/tensor_sparse_dec.h>
6466
#include <tensor_split/gsttensorsplit.h>
6567
#include <tensor_transform/tensor_transform.h>
6668
#include <tensor_if/gsttensorif.h>
@@ -94,6 +96,8 @@ gst_nnstreamer_init (GstPlugin * plugin)
9496
NNSTREAMER_INIT (plugin, reposink, REPOSINK);
9597
NNSTREAMER_INIT (plugin, reposrc, REPOSRC);
9698
NNSTREAMER_INIT (plugin, sink, SINK);
99+
NNSTREAMER_INIT (plugin, sparse_enc, SPARSE_ENC);
100+
NNSTREAMER_INIT (plugin, sparse_dec, SPARSE_DEC);
97101
NNSTREAMER_INIT (plugin, split, SPLIT);
98102
NNSTREAMER_INIT (plugin, transform, TRANSFORM);
99103
NNSTREAMER_INIT (plugin, if, IF);
+9
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
tensor_sparse_sources = [
2+
'tensor_sparse_util.c',
3+
'tensor_sparse_enc.c',
4+
'tensor_sparse_dec.c'
5+
]
6+
7+
foreach s : tensor_sparse_sources
8+
nnstreamer_sources += join_paths(meson.current_source_dir(), s)
9+
endforeach

0 commit comments

Comments
 (0)
Please sign in to comment.