def get_shapes(self): import dace.subsets subsets = dict() for edge in self._state.edges(): if edge.data.data is not None: if "k" in edge.data.subset.free_symbols: continue if edge.data.data not in subsets: subsets[edge.data.data] = edge.data.subset subsets[edge.data.data] = dace.subsets.union( subsets[edge.data.data], edge.data.subset ) bounding_boxes = {name: subset.bounding_box_size() for name, subset in subsets.items()} origins = {name: [r[0] for r in subset.ranges] for name, subset in subsets.items()} return { name: tuple(b + s for b, s in zip(bounding_boxes[name], origins[name])) for name in bounding_boxes.keys() }
def get_shapes(self): import dace.subsets subsets = dict() for edge in self._state.edges(): if edge.data.data is not None: if edge.data.data not in subsets: subsets[edge.data.data] = edge.data.subset subsets[edge.data.data] = dace.subsets.union( subsets[edge.data.data], edge.data.subset) return { name: subset.bounding_box_size() for name, subset in subsets.items() }