pytorch/test/custom_operator/my_custom_ops2.py

11 lines
240 B
Python

import torch
import torch._custom_ops as library
from model import get_custom_op_library_path
torch.ops.load_library(get_custom_op_library_path())
@library.impl_abstract("custom::sin")
def sin_abstract(x):
return torch.empty_like(x)