Minor fixes

This commit is contained in:
Marcos Luciano
2022-02-21 23:46:29 -03:00
parent 66962cfeb8
commit 555152064e
29 changed files with 416 additions and 541 deletions

View File

@@ -101,4 +101,4 @@ nvinfer1::ILayer* activationLayer(
std::abort();
}
return output;
}
}

View File

@@ -29,4 +29,4 @@ nvinfer1::ILayer* channelsLayer(
}
return output;
}
}

View File

@@ -221,4 +221,4 @@ nvinfer1::ILayer* convolutionalLayer(
assert(output != nullptr);
return output;
}
}

View File

@@ -12,4 +12,4 @@ nvinfer1::ILayer* dropoutLayer(
{
nvinfer1::ILayer* output;
return output;
}
}

View File

@@ -27,4 +27,4 @@ nvinfer1::ILayer* implicitLayer(
assert(implicit != nullptr);
return implicit;
}
}

View File

@@ -27,4 +27,4 @@ nvinfer1::ILayer* maxpoolLayer(
pool->setName(maxpoolLayerName.c_str());
return pool;
}
}

View File

@@ -59,4 +59,4 @@ nvinfer1::ILayer* reorgRLayer(
assert(concat != nullptr);
return concat;
}
}

View File

@@ -60,4 +60,4 @@ nvinfer1::ILayer* routeLayer(
}
return output;
}
}

View File

@@ -15,4 +15,4 @@ nvinfer1::ILayer* routeLayer(
std::vector<nvinfer1::ITensor*> tensorOutputs,
nvinfer1::INetworkDefinition* network);
#endif
#endif

View File

@@ -42,4 +42,4 @@ nvinfer1::ILayer* shortcutLayer(
assert(output != nullptr);
return output;
}
}

View File

@@ -21,4 +21,4 @@ nvinfer1::ILayer* upsampleLayer(
std::string layer_name = "upsample_" + std::to_string(layerIdx);
resize_layer->setName(layer_name.c_str());
return resize_layer;
}
}