Need to list out the tfops used in a model/layer

TensorFlow General Discussion
,
1

I have a collection of pre-trained models(.h5) and their training datasets. I need the list of all the Ops used in any layer/model for target-specific model optimization. So far the online solutions have not provided any results. Please help.

OS: UNIX
TF version: 2.5.0

3

You could visualize the TF ops graph with:

1 Like
5

This was really helpful @Bhack . Thanks a lot.
I have one more question to ask.

I need to do the same for tflite models too. Need to have a look at the ops running under the hood for one of my model optimization projects.

Tried using netron, but it shows only weights and abstract level functions. Please point me in the right direction. :slightly_smiling_face:

6

Try with: