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& TvmTransformer::getSupportedOps() { "Reshape", "ReplaceNaN", "Sigmoid", + "Slice", "Softmax", "Split", "Sum",