Overload to allow using binding names instead of indices.
| 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 | { |
| 222 | return engine_get_profile_shape(self, profileIndex, self.getBindingIndex(bindingName.c_str())); |
| 223 | }; |
| 224 | |
| 225 | std::vector<Dims> get_tensor_profile_shape(ICudaEngine& self, std::string const& tensorName, int32_t profileIndex) |
| 226 | { |
nothing calls this directly
no test coverage detected