namespace change

This commit is contained in:
Francesco Gatti
2018-12-14 21:55:16 +01:00
parent 443179359d
commit 6249956469
32 changed files with 374 additions and 374 deletions
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Activation::Activation(Network *net, int act_mode) :
Layer(net) {
@@ -63,4 +63,4 @@ dnnType* Activation::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -2,7 +2,7 @@
#include "Layer.h"
namespace tkDNN {
namespace tk { namespace dnn {
Conv2d::Conv2d( Network *net, int out_ch, int kernelH, int kernelW,
int strideH, int strideW, int paddingH, int paddingW,
@@ -125,4 +125,4 @@ dnnType* Conv2d::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -2,7 +2,7 @@
#include "Layer.h"
namespace tkDNN {
namespace tk { namespace dnn {
Dense::Dense(Network *net, int out_ch, const char* fname_weights) :
LayerWgs(net, net->getOutputDim().tot(), out_ch, 1, 1, 1, fname_weights) {
@@ -55,4 +55,4 @@ dnnType* Dense::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Flatten::Flatten(Network *net) : Layer(net) {
@@ -33,4 +33,4 @@ dnnType* Flatten::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -2,7 +2,7 @@
#include "Layer.h"
namespace tkDNN {
namespace tk { namespace dnn {
Layer::Layer(Network *net) {
@@ -23,4 +23,4 @@ Layer::~Layer() {
checkCUDNN( cudnnDestroyTensorDescriptor(dstTensorDesc) );
}
}
}}
+2 -2
View File
@@ -4,7 +4,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
LayerWgs::LayerWgs(Network *net, int inputs, int outputs,
int kh, int kw, int kl,
@@ -114,4 +114,4 @@ LayerWgs::~LayerWgs() {
}
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
MulAdd::MulAdd(Network *net, dnnType mul, dnnType add) : Layer(net) {
@@ -41,4 +41,4 @@ dnnType* MulAdd::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -5,7 +5,7 @@
#include "Network.h"
#include "Layer.h"
namespace tkDNN {
namespace tk { namespace dnn {
Network::Network(dataDim_t input_dim) {
this->input_dim = input_dim;
@@ -99,4 +99,4 @@ void Network::print() {
}
}
}}
+2 -2
View File
@@ -25,7 +25,7 @@ class Logger : public ILogger {
}
} loggerRT;
namespace tkDNN {
namespace tk { namespace dnn {
std::map<Layer*, nvinfer1::ITensor*>tensors;
@@ -399,4 +399,4 @@ bool NetworkRT::deserialize(const char *filename) {
return true;
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Pooling::Pooling( Network *net, int winH, int winW, int strideH, int strideW,
tkdnnPoolingMode_t pool_mode) :
@@ -107,4 +107,4 @@ dnnType* Pooling::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -9,7 +9,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Region::Region(Network *net, int classes, int coords, int num) :
Layer(net) {
@@ -339,4 +339,4 @@ void RegionInterpret::showImageResult(dnnType *input_h) {
#endif
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Reorg::Reorg(Network *net, int stride) : Layer(net) {
@@ -31,4 +31,4 @@ dnnType* Reorg::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Route::Route(Network *net, Layer **layers, int layers_n) : Layer(net) {
@@ -53,4 +53,4 @@ dnnType* Route::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+2 -2
View File
@@ -3,7 +3,7 @@
#include "Layer.h"
#include "kernels.h"
namespace tkDNN {
namespace tk { namespace dnn {
Softmax::Softmax(Network *net) : Layer(net) {
@@ -44,4 +44,4 @@ dnnType* Softmax::infer(dataDim_t &dim, dnnType* srcData) {
return dstData;
}
}
}}
+1 -1
View File
@@ -53,7 +53,7 @@ public:
virtual void serialize(void* buffer) override {
char *buf = reinterpret_cast<char*>(buffer);
tkDNN::writeBUF(buf, size);
tk::dnn::writeBUF(buf, size);
}
int size;
+1 -1
View File
@@ -8,7 +8,7 @@
class BatchStream
{
public:
BatchStream(tkDNN::dataDim_t dim, int batchSize, int maxBatches)
BatchStream(tk::dnn::dataDim_t dim, int batchSize, int maxBatches)
{
mBatchSize = batchSize;
mMaxBatches = maxBatches;
+6 -6
View File
@@ -74,12 +74,12 @@ public:
virtual void serialize(void* buffer) override {
char *buf = reinterpret_cast<char*>(buffer);
tkDNN::writeBUF(buf, classes);
tkDNN::writeBUF(buf, coords);
tkDNN::writeBUF(buf, num);
tkDNN::writeBUF(buf, c);
tkDNN::writeBUF(buf, h);
tkDNN::writeBUF(buf, w);
tk::dnn::writeBUF(buf, classes);
tk::dnn::writeBUF(buf, coords);
tk::dnn::writeBUF(buf, num);
tk::dnn::writeBUF(buf, c);
tk::dnn::writeBUF(buf, h);
tk::dnn::writeBUF(buf, w);
}
int c, h, w;
+4 -4
View File
@@ -53,10 +53,10 @@ public:
virtual void serialize(void* buffer) override {
char *buf = reinterpret_cast<char*>(buffer);
tkDNN::writeBUF(buf, stride);
tkDNN::writeBUF(buf, c);
tkDNN::writeBUF(buf, h);
tkDNN::writeBUF(buf, w);
tk::dnn::writeBUF(buf, stride);
tk::dnn::writeBUF(buf, c);
tk::dnn::writeBUF(buf, h);
tk::dnn::writeBUF(buf, w);
}
int c, h, w, stride;