Kernels:

kernels-test
/

python-dep

Kernel card Files Files and versions

python-dep / torch-ext /python_dep /__init__.py

danieldk's picture

danieldk HF Staff

Add source files

07906e3 5 days ago

history blame contribute delete

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"]