refactor(//core/lowering): register_const_op ->

register_trt_placeholder_ops Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · May 26, 2020 · 8c26a1b · 8c26a1b
1 parent fa509d9
commit 8c26a1b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/core/lowering/BUILD b/core/lowering/BUILD
@@ -15,7 +15,7 @@ cc_library(
     srcs = [
         "lowering.cpp",
         "drop_unused_nodes.cpp",
-        "register_const_op.cpp"
+        "register_trt_placeholder_ops.cpp"
     ],
     deps = [
         "//core/lowering/passes",

diff --git a/core/lowering/register_const_op.cpp → ...lowering/register_trt_placeholder_ops.cpp b/core/lowering/register_const_op.cpp → ...lowering/register_trt_placeholder_ops.cpp
@@ -7,15 +7,17 @@ c10::AliasAnalysisKind aliasAnalysisFromSchema() {
   return c10::AliasAnalysisKind::FROM_SCHEMA;
 }
 
-/// Op marks a Tensor to be conveted from an Torch Tensor
-/// to a TRT constant Tensor
-RegisterOperators trt_const_op_reg({
+RegisterOperators trt_placeholder_ops_reg({
+  /// Op marks a Tensor to be conveted from an Torch Tensor
+  /// to a TRT constant Tensor
   Operator(
     "trt::const(Tensor val) -> Tensor",
     [](Stack& stack) {
       return 0; //noop
     },
-    aliasAnalysisFromSchema())});
+    aliasAnalysisFromSchema()),
+});
+
 
 } // namespace jit
 } // namespace torch