Kernels:

kernels-test
/

python-dep

Kernel card Files Files and versions

python-dep / build /torch-cpu /__init__.py

danieldk's picture

danieldk HF Staff

Build

4e8294d 5 days ago

179 Bytes

	import torch

	from ._ops import ops
	from .op import _silu_and_mul


	def silu_and_mul(x: torch.Tensor) -> torch.Tensor:
	return ops.silu_and_mul(x)


	__all__ = ["silu_and_mul"]