(node: torch.fx.Node, input_index: int)
| 55 | |
| 56 | |
| 57 | def get_input_node(node: torch.fx.Node, input_index: int) -> torch.fx.Node: |
| 58 | return cast(torch.fx.Node, node.args[input_index]) |
| 59 | |
| 60 | |
| 61 | def normalize_mean_dims(mean_dims: Sequence[int] | int | None, rank: int) -> List[int]: |
no outgoing calls
no test coverage detected