Function get_tensor_profile_shape

python/src/infer/pyCore.cpp:225–232

};

std::vector<Dims> get_tensor_profile_shape(ICudaEngine& self, std::string const& tensorName, int32_t profileIndex)
{
    std::vector<Dims> shapes{};
    shapes.emplace_back(self.getProfileShape(tensorName.c_str(), profileIndex, OptProfileSelector::kMIN));
    shapes.emplace_back(self.getProfileShape(tensorName.c_str(), profileIndex, OptProfileSelector::kOPT));
    shapes.emplace_back(self.getProfileShape(tensorName.c_str(), profileIndex, OptProfileSelector::kMAX));
    return shapes;
};

std::vector<std::vector<int32_t>> engine_get_profile_shape_input(
    ICudaEngine& self, int32_t profileIndex, int32_t bindingIndex)