Я использую TPU через Google Colab и GCP, и хочу сбросить XLA IR. Но я видел xla do c в github xla index , и он показывает путь только тогда, когда бэкэнд - это CPU или GPU.
Я пробовал использовать XLA_FLAGS="--xla_dump_hlo_as_text --xla_dump_to=/content/iir/" TF_XLA_FLAGS=--tf_xla_cpu_global_jit для запуска программа, ориентированная на ЦП, и получить дамп файла hlo. Я также пробовал capture_tpu_file и могу получить только ir для каждого оператора на странице op_profile. Так есть ли способ сбросить XLA IR для всей программы, когда бэкэнд - TPU?
Спасибо!
Jay