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

Method fillAllBindings

samples/common/sampleInference.cpp:127–143  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

125 }
126
127 bool fillAllBindings(int32_t batch, int32_t endBindingIndex)
128 {
129 if (!validateTensorNames(inputs, engine, endBindingIndex))
130 {
131 sample::gLogError << "Invalid tensor names found in --loadInputs flag." << std::endl;
132 return false;
133 }
134 for (int32_t b = 0; b < endBindingIndex; b++)
135 {
136 TensorInfo tensorInfo;
137 tensorInfo.bindingIndex = b;
138 getTensorInfo(tensorInfo);
139 tensorInfo.updateVolume(batch);
140 fillOneBinding(tensorInfo);
141 }
142 return true;
143 }
144
145 void getTensorInfo(TensorInfo& tensorInfo);
146

Callers

nothing calls this directly

Calls 2

validateTensorNamesFunction · 0.85
updateVolumeMethod · 0.80

Tested by

no test coverage detected