Flops calculation in tensorflow 2.x version.

[weichen456]

Please make sure that this is a feature request. As per our GitHub Policy, we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:feature_template

System information

• TensorFlow version (you are using):
  tensorflow 2.2
• Are you willing to contribute it (Yes/No):
  No

Describe the feature and the current behavior/state.
Flops calculation in tf1.x usually use tf.compat.v1.profiler.ProfileOptionBuilder,but it doesn't work success in tf2.x,So,anyone can show me the official example to calculate flop in tf2.x version?
Will this change the current api? How?
No
Who will benefit with this feature?
People who care about model efficiency.
Any Other info.

[amahendrakar]

Flops calculation in tf1.x usually use tf.compat.v1.profiler.ProfileOptionBuilder,but it doesn't work success in tf2.x

@weichen456,
Could you please provide a minimal reproducible code regarding this so that we can look into it.

Also, please check this documentation for ProfileOptionBuilder and let us know if it helps. Thanks!

[weichen456]

Flops calculation in tf1.x usually use tf.compat.v1.profiler.ProfileOptionBuilder,but it doesn't work success in tf2.x

@weichen456,
Could you please provide a minimal reproducible code regarding this so that we can look into it.

Also, please check this documentation for ProfileOptionBuilder and let us know if it helps. Thanks!

hello,
my code is here https://colab.research.google.com/drive/1pA_cYnMj5JXxCCsJcGISmaZjmXJ5BTf9?usp=sharing
The params is work well, but the flops is zero.
can you give me a official example in tf2.x version?

[amahendrakar]

Was able to reproduce the issue with TF v2.2 and TF-nightly.

Works as expected with TF v1.15. Please find the attached gist. Thanks!

[henglicad]

It will be very valuable to have the issue fixed.

[bhack]

ThIs Is a duplicate of #32809

[sushreebarsa]

@weichen456 this ticket is a duplicate of #32809 ,please let us know if we can close this issue ..thanks !

[weichen456]

@weichen456 this ticket is a duplicate of #32809 ,please let us know if we can close this issue ..thanks !

Ok

[sushreebarsa]

@weichen456 Thank you for your response. closing this issue.

[google-ml-butler]

Are you satisfied with the resolution of your issue?
Yes
No