f1b19a6b67f353762025dae38c2ab9363f553930,python/dgl/heterograph.py,DGLHeteroGraph,to,#DGLHeteroGraph#Any#,4050

Before Change

        for i in range(len(self._node_frames)):
            for k in self._node_frames[i].keys():
                self._node_frames[i][k] = F.copy_to(self._node_frames[i][k], ctx)
        for i in range(len(self._edge_frames)):
            for k in self._edge_frames[i].keys():
                self._edge_frames[i][k] = F.copy_to(self._edge_frames[i][k], ctx)
        return self

    def local_var(self):
        Return a heterograph object that can be used in a local function scope.

After Change

            new_nframes.append(FrameRef(Frame(new_feats)))
        new_eframes = []
        for eframe in self._edge_frames:
            new_feats = {k : F.copy_to(feat, ctx) for k, feat in eframe.items()}
            new_eframes.append(FrameRef(Frame(new_feats)))
        // TODO(minjie): replace the following line with the commented one to enable GPU graph.
        new_gidx = self._graph
        //new_gidx = self._graph.copy_to(utils.to_dgl_context(ctx))

Instances