Do profiling inside while thunks and conditionals.

We now look into the computations of kWhile and kConditional ops when profiling.
This still does not help regarding the statistics of the estimated optimum,
but at least we can see the relative performance of the ops within a
subcomputation.

PiperOrigin-RevId: 202916616