Returns true if the given node is a get attr node for a tensor of the model
(node: torch.fx.Node)
| 90 | |
| 91 | |
| 92 | def is_get_attr_node(node: torch.fx.Node) -> bool: |
| 93 | """ |
| 94 | Returns true if the given node is a get attr node for a tensor of the model |
| 95 | """ |
| 96 | return isinstance(node, torch.fx.Node) and node.op == "get_attr" |
| 97 | |
| 98 | |
| 99 | def is_param_node(exp_prog: ExportedProgram, node: torch.fx.Node) -> bool: |
no outgoing calls
no test coverage detected