python/featgraph/module/sddmm.py [37:43]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self._num_col_partitions = num_col_partitions
        # To be updated in self.register
        self._target = None
        self._ctx = None
        self._compute_func = None
        self._schedule_func = None
        self._register()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



python/featgraph/module/spmm.py [33:39]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self._num_col_partitions = num_col_partitions
        # To be updated in self._register
        self._target = None
        self._ctx = None
        self._compute_func = None
        self._schedule_func = None
        self._register()
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -