commit bd9617d5af0c1cfb4badfaa92c240ed6b9b08c4e
author Hao Lu <hlu@fb.com> Mon May 04 01:27:55 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Mon May 04 01:30:06 2020 -0700
tree 902d6f698de7e78496fb8b639eccc1c344cab2c6
parent 843c0230f2928aad61a6940688da3a6cd6c4cd57

[TVM] Implement UnPackRecordsOp (#37489)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37489

Reviewed By: yinghai

Differential Revision: D21196596

fbshipit-source-id: 58b8bc3afc472e02ef7e3d31151fe8ace2be2a73

caffe2/opt/tvm_transformer.cc

diff --git a/caffe2/opt/tvm_transformer.cc b/caffe2/opt/tvm_transformer.cc
index a32e836..25a88a7 100644
--- a/caffe2/opt/tvm_transformer.cc
+++ b/caffe2/opt/tvm_transformer.cc
@@ -199,15 +199,32 @@
 
 const std::unordered_set<std::string>& TvmTransformer::getSupportedOps() {
   const static std::unordered_set<std::string> supported_ops{
-      "Add",        "BatchGather", "BatchMatMul",
-      "Cast",       "Clip",        "Concat",
-      "Copy",       "DotProduct",  "EnsureCPUOutput",
-      "ExpandDims", "FC",          "FCTransposed",
-      "Flatten",    "Logit",       "MatMul",
-      "Mul",        "Relu",        "Reshape",
-      "ReplaceNaN", "Sigmoid",     "Softmax",
-      "Split",      "Sum",         "Tanh",
+      "Add",
+      "BatchGather",
+      "BatchMatMul",
+      "Cast",
+      "Clip",
+      "Concat",
+      "Copy",
+      "DotProduct",
+      "EnsureCPUOutput",
+      "ExpandDims",
+      "FC",
+      "FCTransposed",
+      "Flatten",
+      "Logit",
+      "MatMul",
+      "Mul",
+      "Relu",
+      "Reshape",
+      "ReplaceNaN",
+      "Sigmoid",
+      "Softmax",
+      "Split",
+      "Sum",
+      "Tanh",
       "Transpose",
+      "UnPackRecords",
   };
   return supported_ops;
 }