| 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 |
nothing calls this directly
no test coverage detected