1 | #pragma once |
---|---|
2 | |
3 | #include <c10/macros/Export.h> |
4 | |
5 | namespace c10 { |
6 | struct OperatorName; |
7 | } |
8 | |
9 | namespace at { |
10 | |
11 | // check if an op is a custom op (i.e. did not come from native_functions.yaml) |
12 | TORCH_API bool is_custom_op(const c10::OperatorName& opName); |
13 | } |
14 |