(self, sort, node)
| 551 | self.flattened = None |
| 552 | |
| 553 | def add(self, sort, node): |
| 554 | self.names[sort] = _tensor_name_base(node.name) |
| 555 | self.nodes[sort] = node |
| 556 | |
| 557 | def flatten_nodes(self): |
| 558 | """Return a list of all the node protos in aggregation sorted order.""" |