summaryrefslogtreecommitdiff
path: root/caffe2/core
diff options
context:
space:
mode:
Diffstat (limited to 'caffe2/core')
-rw-r--r--caffe2/core/c10_operator.h3
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));