Add custom-call ops to roofline.

Commit

1 year ago

Add custom-call ops to roofline. Recursively calculates the roofline result for the primitives from the custom function. PiperOrigin-RevId: 772252912