Refactoring & documentation

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-08 11:20:48 +02:00
parent 326c7e0940
commit ae1d8cd9e6
9 changed files with 133 additions and 168 deletions
+1 -4
View File
@@ -24,11 +24,8 @@ Reshape::~Reshape() {
dnnType* Reshape::infer(dataDim_t &dim, dnnType* srcData) {
//transpose per channel
//just copies the data and changes the output dim
checkCuda( cudaMemcpy(dstData, srcData, dim.n*dim.c*dim.h*dim.w*sizeof(dnnType), cudaMemcpyDeviceToDevice));
//update data dimensions
dim = output_dim;
return dstData;