Method argument

python/cutlass_cppgen/backend/frontend.py:101–109

(tensor, is_output=False)

    @staticmethod
    def argument(tensor, is_output=False):
        if is_numpy_tensor(tensor):
            return NumpyFrontend.argument(tensor, is_output)
        elif is_torch_tensor(tensor):
            return TorchFrontend.argument(tensor)
        elif is_cupy_tensor(tensor):
            return CupyFrontend.argument(tensor)
        else:
            raise NotImplementedError("Unknown Tensor Type")