(x)
| 2178 | with ops.control_dependencies(new_summaries): |
| 2179 | |
| 2180 | def map_fn(x): |
| 2181 | # TODO(apassos) figure out how to trigger with tensor arrays as well |
| 2182 | if isinstance(x, tensor_array_ops.TensorArray): |
| 2183 | return x |
| 2184 | return array_ops.identity(x) |
| 2185 | |
| 2186 | body_result = nest.map_structure( |
| 2187 | map_fn, body_result, expand_composites=True) |