From 378c8e983773eb171a9a9ca3969aec946a4cfbec Mon Sep 17 00:00:00 2001 From: Leyuan Wang Date: Fri, 31 May 2019 00:14:43 -0700 Subject: [PATCH] alleviate hanging issue caused by concat op --- src/relay/op/tensor/transform.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc index 873e75d9660bc..16fb8bdca55f9 100644 --- a/src/relay/op/tensor/transform.cc +++ b/src/relay/op/tensor/transform.cc @@ -6,9 +6,9 @@ * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at - * + * * http://www.apache.org/licenses/LICENSE-2.0 - * + * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY @@ -292,7 +292,7 @@ RELAY_REGISTER_OP("concatenate") .add_type_rel("Concatenate", ConcatenateRel) .set_attr("FInferCorrectLayout", ConcatenateLayout) .set_attr("FTVMCompute", ConcatenateCompute) -.set_attr("TOpPattern", kInjective); +.set_attr("TOpPattern", kOpaque); TVM_REGISTER_NODE_TYPE(StackAttrs);