(node: Any)
| 265 | |
| 266 | |
| 267 | def is_tensor_arg_node(node: Any) -> bool: |
| 268 | if isinstance(node, torch.fx.Node): |
| 269 | return is_tensor_node(node) |
| 270 | elif isinstance(node, (list, tuple)): |
| 271 | if len(node) == 0: |
| 272 | return False |
| 273 | return all(is_tensor_node(n) for n in node) |
| 274 | |
| 275 | return False |
| 276 | |
| 277 | |
| 278 | def num_tensor_arg_nodes(node: torch.fx.Node) -> int: |
no test coverage detected