|View source on GitHub|
Removes all the mapped or applied identity functions in
tff.framework.remove_mapped_or_applied_identity( comp )
This transform traverses
comp postorder, matches the following pattern, and
removes all the mapped or applied identity fucntions by replacing the
Call / \
Intrinsic(<(x -> x), y>)
with its argument:
comp: The computation building block in which to perform the removals.
A new computation with the transformation applied or the original
TypeError: If types do not match.