| 209 | }; |
| 210 | |
| 211 | std::vector<Dims> engine_get_profile_shape(ICudaEngine& self, int32_t profileIndex, int32_t bindingIndex) |
| 212 | { |
| 213 | std::vector<Dims> shapes{}; |
| 214 | shapes.emplace_back(self.getProfileDimensions(bindingIndex, profileIndex, OptProfileSelector::kMIN)); |
| 215 | shapes.emplace_back(self.getProfileDimensions(bindingIndex, profileIndex, OptProfileSelector::kOPT)); |
| 216 | shapes.emplace_back(self.getProfileDimensions(bindingIndex, profileIndex, OptProfileSelector::kMAX)); |
| 217 | return shapes; |
| 218 | }; |
| 219 | // Overload to allow using binding names instead of indices. |
| 220 | std::vector<Dims> engine_get_profile_shape_str(ICudaEngine& self, int32_t profileIndex, std::string const& bindingName) |
| 221 | { |
no test coverage detected