From 06c277f38e8da2688381bdcacf4a5e350b38d529 Mon Sep 17 00:00:00 2001
From: Hao Lu <haolu7@gmail.com>
Date: Tue, 1 Sep 2020 12:32:09 -0700
Subject: [PATCH] [TVM] Support slice op (#43969)

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

Reviewed By: yinghai

Differential Revision: D23413340

fbshipit-source-id: 20168bd573b81ce538e3589b72aba9590c3c055e
---
 caffe2/opt/tvm_transformer.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/caffe2/opt/tvm_transformer.cc b/caffe2/opt/tvm_transformer.cc
index 945f31eca0c..2c62f6521b0 100644
--- a/caffe2/opt/tvm_transformer.cc
+++ b/caffe2/opt/tvm_transformer.cc
@@ -204,6 +204,7 @@ const std::unordered_set<std::string>& TvmTransformer::getSupportedOps() {
       "Reshape",
       "ReplaceNaN",
       "Sigmoid",
+      "Slice",
       "Softmax",
       "Split",
       "Sum",