keras-flops (pypi): FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)
Keywords: tensorflow2|flops|profiler