diff options
Diffstat (limited to 'caffe2/core')
-rw-r--r-- | caffe2/core/c10_operator.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/caffe2/core/c10_operator.h b/caffe2/core/c10_operator.h index 240a16be25..86d8911a1c 100644 --- a/caffe2/core/c10_operator.h +++ b/caffe2/core/c10_operator.h @@ -1,6 +1,7 @@ #pragma once #include <ATen/core/function_schema.h> +#include <ATen/core/interned_strings.h> #include <ATen/core/op_registration/op_registration.h> #include <vector> @@ -97,7 +98,7 @@ inline c10::FunctionSchema make_function_schema_for_c10(const char* OperatorName IValue()); return c10::FunctionSchema( - std::string("_caffe2::") + OperatorName, + Symbol::caffe2(OperatorName).toQualString(), "", std::move(actual_inputs), std::move(outputs)); |