def parse_tff_to_tf(comp): comp, _ = tree_transformations.insert_called_tf_identity_at_leaves(comp) parser_callable = tree_to_cc_transformations.TFParser() comp, _ = tree_transformations.replace_called_lambda_with_block(comp) comp, _ = tree_transformations.inline_block_locals(comp) comp, _ = tree_transformations.replace_selection_from_tuple_with_element(comp) new_comp, transformed = transformation_utils.transform_postorder( comp, parser_callable) return new_comp, transformed
def transformation_fn(x): x, _ = tree_transformations.remove_mapped_or_applied_identity(x) x, _ = tree_transformations.inline_block_locals(x) x, _ = tree_transformations.replace_selection_from_tuple_with_element(x) return x