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()
}