Есть ли аналог функции tf.compat.v1.wrap_function на tensorflow для с++?

Есть граф некой модели, который получается нормально запустить на Python, однако модель имеет выход произвольный длины, как бы вообще tensorflow должна сама понимать размерность выхода и подгонять, однако в конкретно в этом случае, размеры подгоняются дополнительной оберткой над функцией(проверял в отладке).

mtcnn_fun2 = tf.compat.v1.wrap_function(mtcnn_fun, [
tf.TensorSpec(shape=[None, None, 3], dtype=tf.float32),
tf.TensorSpec(shape=[], dtype=tf.float32),
tf.TensorSpec(shape=[], dtype=tf.float32),
tf.TensorSpec(shape=[3], dtype=tf.float32)
])

Без этой функции, нормально размер выходных тензоров на Python не корректируется. Не могу найти аналог этой функции для с++ версии, чтобы и там тоже скорректировать выход сетки.