registering all functions with torch.{xxx} dispatcher