Tensorflow - XLA | Как TF ops понижен до XLA для обучения
Как тензорные операции снижаются до операций XLA во время тренировок?
Файлы в тензорном потоке / компиляторе /tf2xla/kernels определяют только прямой проход / вывод в терминах операций в ComputationBuilder.
Этот вопрос относится к " Tensorflow - XLA | Передача тензоров внешним функциям во время выполнения", в которой мне нужны операции TF, которые понижены до операций XLA, чтобы передавать тензоры во время обучения и вывода. Пожалуйста, поделитесь своими мыслями и по другому вопросу.
Спасибо
1 ответ
Насколько я понимаю, Tensorflow вычисляет производные, необходимые для обратного распространения, прежде чем передать график в tf2xla для понижения. Все операции Tensorflow для обучения и логического вывода снижены до XLA с использованием tf2xla; другого механизма нет.