Предоставляет ли TensorFlow или XLA API Python для чтения и анализа дампа модуля MHLO mlir? ⇐ Python

[Anonymous]

Я включил XLA при запуске TensorFLow и для дальнейшей оптимизации объединенных ядер добавил экспорт XLA_FLAGS="--xla_dump_to=/tmp/xla_dump" и получил дамп IR, включая lmhlo. xxx.mlir и другие llvm IR.
Теперь, когда я пытаюсь проанализировать эти дампы mlir, я хочу их прочитать. как структурированные модули MLIR, поэтому мне нужно правильно его прочитать и проанализировать. Но я не могу найти никаких ресурсов, документирующих, как это сделать на чистом Python. Я попробовал модуль «pymlir», но он не очень хорошо работает с этим модулем TF XLA HLO, возможно, сброшенный модуль имеет другой формат.
Как и кто-нибудь? умеешь читать и парсить этот сваленный млир?

Подробнее здесь: https://stackoverflow.com/questions/791 ... ed-mhlo-ml