Searched defs:op_impl (Results 1 – 3 of 3) sorted by relevance
127 def impl_decorator(op_impl): argument752 def impl_decorator(op_impl): argument
645 def op_impl(inputs: List[torch.Tensor]): function
2509 def registerOp(cls, op_key, full_schema, op_impl, dispatch_key): argument