Checks if the given tensor is a graph input Args: tensor: EdgeIR Tensor that is being checked for graph input
(self, tensor: torch.fx.Node)
| 105 | return self._exported_program |
| 106 | |
| 107 | def is_graph_input(self, tensor: torch.fx.Node) -> bool: |
| 108 | """ |
| 109 | Checks if the given tensor is a graph input |
| 110 | |
| 111 | Args: |
| 112 | tensor: EdgeIR Tensor that is being checked for graph input |
| 113 | """ |
| 114 | return tensor.op == "placeholder" and not is_param_node( |
| 115 | self.exported_program, tensor |
| 116 | ) |
| 117 | |
| 118 | def is_graph_output(self, tensor: torch.fx.Node) -> bool: |
| 119 | """ |
no test coverage detected