1from model import get_custom_op_library_path 2 3import torch 4 5 6torch.ops.load_library(get_custom_op_library_path()) 7 8 9@torch.library.impl_abstract("custom::sin") 10def sin_abstract(x): 11 return torch.empty_like(x) 12
1from model import get_custom_op_library_path 2 3import torch 4 5 6torch.ops.load_library(get_custom_op_library_path()) 7 8 9@torch.library.impl_abstract("custom::sin") 10def sin_abstract(x): 11 return torch.empty_like(x) 12