mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

flatten implemented

Browse Source
This commit is contained in:
Francesco Gatti
2017-06-29 09:24:06 +00:00
parent a04c3eaf88
commit 97226cbc78
7 changed files with 69 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test.cpp
+2
View File
@@ -18,6 +18,7 @@ int main() {
tkDNN::Activation a0 (&net, c0.output_dim, tkDNN::ACTIVATION_RELU);
tkDNN::Conv3d c1 (&net, a0.output_dim, 4, 2, 2, 2, 1, 1, 1, c1_bin, c1_bias_bin);
tkDNN::Activation a1 (&net, c1.output_dim, tkDNN::ACTIVATION_ELU);
tkDNN::Flatten f1 (&net, a1.output_dim);
// Load input
value_type *data;
@@ -33,6 +34,7 @@ int main() {
data = a0.infer(dim, data); dim.print();
data = c1.infer(dim, data); dim.print();
data = a1.infer(dim, data); dim.print();
data = f1.infer(dim, data); dim.print();
TIMER_STOP
// Print result