Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand


Counts and returns model FLOPs.

model A model instance.
inputs_kwargs An optional dictionary of argument pairs specifying inputs' shape specifications to getting corresponding concrete function.
output_path A file path to write the profiling results to.

The model's FLOPs.