tff.framework.replace_called_lambda_with_block

View source on GitHub

Replaces all the called lambdas in comp with a block.

tff.framework.replace_called_lambda_with_block(
    comp
)