Method gen_key

python/oneflow/nn/graph/cache.py:222–228

(self, *args, **kwargs)

                )

    def gen_key(self, *args, **kwargs):
        flattened_shapes = []
        args_tree = ArgsTree((args, kwargs), False)
        for arg in args_tree.iter_nodes():
            if isinstance(arg, Tensor):
                flattened_shapes.append(arg.shape)
        return tuple(flattened_shapes)

    def get_graph(self, *args, **kwargs):
        if self._cache is None: