index d2d2c51e3b489379f3d6dd703b7adf296747e670..c1576ce8a8d156c3f539ce02f29a86b69c5dd3be 100644 (file)
}
// setup input layer
- newLayer = malloc(sizeof(timlCNNLayer));
- if (newLayer == NULL) {
+ if (timlUtilMallocHost((void**)&newLayer, sizeof(timlCNNLayer))) {
return ERROR_CNN_LAYER_ALLOCATION;
}
newLayer->channel = featureMapChannel;
newLayer->row = featureMapRow;
newLayer->col = featureMapCol;
+ newLayer->batchSize = cnn->params.batchSize;
+ newLayer->maxBatchSize = cnn->params.maxBatchSize;
newLayer->inputParams = params;
if (newLayer->inputParams.row == -1) {
if (newLayer->inputParams.channel == -1) {
newLayer->inputParams.channel = newLayer->channel;
}
+ newLayer->inputParams.shared = false;
+ newLayer->inputParams.inputData = NULL;
newLayer->featureMap = NULL;
newLayer->delta = NULL;