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

Function optionalWeights

python/src/infer/pyGraph.cpp:36–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

34 namespace lambdas
35 {
36 Weights optionalWeights(Weights* weights)
37 {
38 if (weights)
39 {
40 return *weights;
41 }
42 return Weights{DataType::kFLOAT, nullptr, 0};
43 }
44
45 static const auto get_dynamic_range = [] (ITensor const& self) -> py::object {
46 if (self.dynamicRangeIsSet()) {

Callers 4

add_convolutionFunction · 0.85
pyGraph.cppFile · 0.85
add_fully_connectedFunction · 0.85
add_deconvolutionFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected