diff --git a/src/Builder/FrontendDialectTransformer.cpp b/src/Builder/FrontendDialectTransformer.cpp index d7f852c..c806eef 100644 --- a/src/Builder/FrontendDialectTransformer.cpp +++ b/src/Builder/FrontendDialectTransformer.cpp @@ -313,26 +313,6 @@ private: node, inputs, expectedNumOperands, expectedNumResults); } - void ImportNodeReshape(onnx::NodeProto node) { - int expectedNumOperands = mlir::ONNXReshapeOp::getNumberOfOperands(); - int expectedNumResults = mlir::ONNXReshapeOp::getNumberOfResults(); - std::vector inputs; - std::string item; - for (int i = 0; i < node.input().size(); ++i) { - item = node.input()[i]; - // For the second argument, check if there exists an initializer. - if (initializedTensors.ContainKey(legalize_name(item))) { - inputs.push_back(initializedTensors.EmitInitializerForInputTensor( - UnknownLoc(), builder_, legalize_name(item))); - } else if (frontend_symbols_.ContainKey(legalize_name(item))) { - inputs.push_back(frontend_symbols_.GetTensorByOnnxName(item)); - } - } - - buildOutputAndOperation( - node, inputs, expectedNumOperands, expectedNumResults); - } - /*! * Special handle for MaxPool operations. */ diff --git a/src/Builder/OpBuildTable.inc b/src/Builder/OpBuildTable.inc index 3e842fd..8517f82 100644 --- a/src/Builder/OpBuildTable.inc +++ b/src/Builder/OpBuildTable.inc @@ -223,7 +223,7 @@ if (opName == "ReduceSumSquare") if (opName == "Relu") buildOperation(node); if (opName == "Reshape") - ImportNodeReshape(node); + buildOperation(node); if (opName == "Resize") buildOperation(node); if (opName == "ReverseSequence") diff --git a/utils/gen_onnx_mlir.py b/utils/gen_onnx_mlir.py index e81ba66..393c266 100644 --- a/utils/gen_onnx_mlir.py +++ b/utils/gen_onnx_mlir.py @@ -239,7 +239,6 @@ special_op_handler = dict([ ("MaxPool", "ImportNodeMaxPool"), ("BatchNormalization", "ImportNodeBatchNormalization"), ("Pad", "ImportNodePad"), - ("Reshape", "ImportNodeReshape"), #("Transpose", "ImportNodeTranspose") ])