(leaf_node)
| 367 | ) |
| 368 | |
| 369 | def leaf_node_fn(leaf_node): |
| 370 | arg = leaf_node.value() |
| 371 | name = leaf_node.prefix() + "_" + leaf_node.name() |
| 372 | is_tensor, repr_str = self.__io_tensor_check_and_gen(arg, io_type, name) |
| 373 | if is_tensor: |
| 374 | self.__print( |
| 375 | 0, |
| 376 | 1, |
| 377 | f"{repr_str} is a Tensor, {func_desc} transformation has been done.", |
| 378 | ) |
| 379 | return map_tensor(arg) |
| 380 | else: |
| 381 | self.__print( |
| 382 | 0, |
| 383 | 0, |
| 384 | f"{repr_str} is not a Tensor, {func_desc} transformation will be ignored.", |
| 385 | ) |
| 386 | return arg |
| 387 | |
| 388 | out = args_tree.map_leaf(leaf_node_fn) |
| 389 | mapped_args = out[0] |
nothing calls this directly
no test coverage detected