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

Function add_fully_connected

python/src/infer/pyGraph.cpp:122–125  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

120 };
121
122 IFullyConnectedLayer* add_fully_connected(INetworkDefinition& self, ITensor& input, int32_t numOutputs, Weights kernel, Weights* bias)
123 {
124 return self.addFullyConnected(input, numOutputs, kernel, optionalWeights(bias));
125 };
126
127 IGridSampleLayer* add_grid_sample(INetworkDefinition& self, ITensor& input, ITensor& grid)
128 {

Callers

nothing calls this directly

Calls 2

optionalWeightsFunction · 0.85
addFullyConnectedMethod · 0.80

Tested by

no test coverage detected