(
nodes: Iterable[Node], graph_signature: Optional[ExportGraphSignature]
)
| 498 | |
| 499 | |
| 500 | def _get_mutable_buffer_specs( |
| 501 | nodes: Iterable[Node], graph_signature: Optional[ExportGraphSignature] |
| 502 | ) -> Set[TensorSpec]: |
| 503 | mutable_buffer_specs: Set[TensorSpec] = set() |
| 504 | for node in nodes: |
| 505 | if _is_mutable_buffer(node, graph_signature): |
| 506 | for spec in get_node_tensor_specs(node): |
| 507 | mutable_buffer_specs.add(spec) |
| 508 | return mutable_buffer_specs |
| 509 | |
| 510 | |
| 511 | def _do_user_inputs_exist(graph_signature: Optional[ExportGraphSignature]) -> bool: |
no test coverage detected