def run()

in python/featgraph/module/spmm.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def run(self, input_tvm_ndarrays):
        """Run the built func with the given inputs input_tvm_ndarrays.

        Parameters
        ----------
        input_tvm_ndarrays : list of tvm.ndarray
            The required input tvm ndarrays other than adj

        Returns
        -------
        self.out_tvm: tvm.ndarray
            The output tvm ndarray
        """
        if self._num_col_partitions > 1:
            self._func(*input_tvm_ndarrays, self._adj_s1_pos_tvm, self._adj_s1_idx_tvm, \
                self._adj_vals_tvm, self.out_tvm)
        else:
            self._func(*input_tvm_ndarrays, self._adj_indptr_tvm, self._adj_indices_tvm, \
                self._adj_vals_tvm, self.out_tvm)
        return self.out_tvm