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

Function add_deconvolution

python/src/infer/pyGraph.cpp:152–155  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

150 };
151
152 IDeconvolutionLayer* add_deconvolution(INetworkDefinition& self, ITensor& input, int32_t numOutputMaps, DimsHW kernelSize, Weights kernel, Weights* bias)
153 {
154 return self.addDeconvolution(input, numOutputMaps, kernelSize, kernel, optionalWeights(bias));
155 };
156
157 static const auto add_scatter = [](INetworkDefinition& self, ITensor& data, ITensor& indices, ITensor& updates, ScatterMode mode)
158 {

Callers

nothing calls this directly

Calls 2

optionalWeightsFunction · 0.85
addDeconvolutionMethod · 0.80

Tested by

no test coverage detected