(
nodes: Iterable[Node], graph_signature: Optional[ExportGraphSignature] = None
)
| 532 | |
| 533 | |
| 534 | def get_graph_input_tensors( |
| 535 | nodes: Iterable[Node], graph_signature: Optional[ExportGraphSignature] = None |
| 536 | ) -> Set[TensorSpec]: |
| 537 | graph_input_tensors = set() |
| 538 | for node in nodes: |
| 539 | if node.op == "placeholder" and not _is_mutable_buffer(node, graph_signature): |
| 540 | for spec in get_node_tensor_specs(node): |
| 541 | graph_input_tensors.add(spec) |
| 542 | |
| 543 | return graph_input_tensors |
| 544 | |
| 545 | |
| 546 | def get_graph_output_tensors(nodes: Iterable[Node]) -> Set[TensorSpec]: |
no test coverage detected