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

MulAdd implemented

Browse Source
This commit is contained in:
Francesco Gatti
2017-06-29 10:06:49 +00:00
parent 97226cbc78
commit cc99347560
6 changed files with 82 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test.cpp
+2
View File
@@ -19,6 +19,7 @@ int main() {
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);
tkDNN::MulAdd m1 (&net, f1.output_dim, 2, 1);
// Load input
value_type *data;
@@ -35,6 +36,7 @@ int main() {
data = c1.infer(dim, data); dim.print();
data = a1.infer(dim, data); dim.print();
data = f1.infer(dim, data); dim.print();
data = m1.infer(dim, data); dim.print();
TIMER_STOP
// Print result