Skip to content

Commit a2834ad

Browse files
committed
Import op_registration directly
1 parent acec434 commit a2834ad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/csrc/io/image/image.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#include "image.h"
22

3+
#include <ATen/core/op_registration/op_registration.h>
34
#include <Python.h>
4-
#include <torch/library.h>
55

66
// If we are in a Windows environment, we need to define
77
// initialization functions for the _custom_ops extension

0 commit comments

Comments
 (0)