Counting FLOPS in tensorflow

112 views Asked by At

Is there a way to count FLOPS for the training and prediction of tensorflow models?
The models are running on a CPU using tensorflow 2.8.0 and i would not like to use an external (e.g. command line) tool.

0

There are 0 answers