[PATCH] D81515: [llvm] Release-mode ML InlineAdvisor

Gaurav Jain via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Wed Oct 21 16:04:09 PDT 2020


gjain added a comment.

In D81515#2344814 <https://reviews.llvm.org/D81515#2344814>, @mtrofin wrote:

> In D81515#2344805 <https://reviews.llvm.org/D81515#2344805>, @AmirJamez wrote:
>
>> Would you provide scripts to load the model and see the layers?
>
> Re. second question, visualization - this is a question for Yundi, Gaurav, or Eugene (they are the ML experts). I'll venture "tensorboard" as an answer, but I'll make sure they give the authoritative one in a moment.

You should be able to use tensorboard but you need to first import the model into tensorboard with https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/import_pb_to_tensorboard.py. Something like `python import_pb_to_tensorboard.py --model_dir=llvm/lib/Analysis/models/inliner/ --log_dir=/tmp/inliner` should work. Then you'll be able to run tensorboard on the log_dir.

Here's a hosted visualization from tensorboard for your convenience: https://tensorboard.dev/experiment/C45o0HjZTPGRSqpOrdkbeg/#graphs


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D81515/new/

https://reviews.llvm.org/D81515



More information about the llvm-commits mailing list