Add TKDNN_MODE variable to the name of the network.

This commit permits to obtain different .rt files for different
precision optimizations of the same network.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-03-30 15:08:25 +02:00
parent 42240de49a
commit 0474f019cf
26 changed files with 53 additions and 24 deletions
+1 -1
View File
@@ -22,7 +22,7 @@ int main() {
tk::dnn::Dense l6(&net, 10, d3_bin);
tk::dnn::Softmax l7(&net);
tk::dnn::NetworkRT netRT(&net, "mnist.rt");
tk::dnn::NetworkRT netRT(&net, net.getNetworkRTName("mnist"));
// Load input
dnnType *data;