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

Added num of layers in net and netRT,suggestions from pull request 270 on main repo

Browse Source
This commit is contained in:
perseusdg
2021-11-14 07:13:42 +05:30
parent 367061fea2
commit 6f936096ae
3 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+5 -1
View File
@@ -183,7 +183,11 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
output_dim.h = oDim.d[1];
output_dim.w = oDim.d[2];
output_dim.print();
if(builderActive){
std::cout<<"NUMBER OF LAYERS IN NETWORK : "<<networkRT->getNbLayers()<<std::endl;
}
std::cout<<"NUMBER OF LAYERS IN ENGINE : "<<engineRT->getNbLayers()<<std::endl;
// create GPU buffers and a stream
for(int i=0; i<engineRT->getNbBindings(); i++) {
Dims dim = engineRT->getBindingDimensions(i);