MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / add_convolution

Function add_convolution

python/src/infer/pyGraph.cpp:112–115  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

110 };
111
112 IConvolutionLayer* add_convolution(INetworkDefinition& self, ITensor& input, int32_t numOutputMaps, DimsHW kernelSize, Weights kernel, Weights* bias)
113 {
114 return self.addConvolution(input, numOutputMaps, kernelSize, kernel, optionalWeights(bias));
115 };
116
117 static const auto add_convolution_nd = [](INetworkDefinition& self, ITensor& input, int32_t numOutputMaps, Dims kernelSize, Weights kernel, Weights* bias)
118 {

Callers

nothing calls this directly

Calls 2

optionalWeightsFunction · 0.85
addConvolutionMethod · 0.80

Tested by

no test coverage detected