| 30 | |
| 31 | |
| 32 | class DataFlow(object): |
| 33 | def __init__(self, n_id): |
| 34 | self.n_id = n_id |
| 35 | self.__last_n_id__ = n_id |
| 36 | self.blocks = [] |
| 37 | |
| 38 | def append(self, n_id, res_n_id, e_id, edge_index): |
| 39 | size = [tf.shape(self.__last_n_id__)[0], tf.shape(n_id)[0]] |
| 40 | block = Block(n_id, res_n_id, e_id, edge_index, size) |
| 41 | self.blocks.append(block) |
| 42 | self.__last_n_id__ = n_id |
| 43 | |
| 44 | def __len__(self): |
| 45 | return len(self.blocks) |
| 46 | |
| 47 | def __getitem__(self, idx): |
| 48 | return self.blocks[::-1][idx] |
| 49 | |
| 50 | def __iter__(self): |
| 51 | for block in self.blocks[::-1]: |
| 52 | yield block |
no outgoing calls
no test coverage detected