Я кодирую пользовательский оценщик tf и хочу скомпилировать обученный оценщик и использовать его в другом приложении.
В моей функции model_fn
Я хочу иметь алгоритм во время поезда / оценки / прогнозированияцикл и другой эквивалентный алгоритм, когда я компилирую .so
(действительно, некоторые операции, которые я использую во время обучения, не имеют каких-либо целевых операций в среде xla, поэтому я заменяю их эквивалентными операциями после завершения обучения).
Как этого достичь?