colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove ImportNodeReshape (#208)

This commit is contained in:
Tung D. Le 2020-07-08 23:27:14 +09:00 committed by GitHub
parent 01a4977c74
commit 0a936edf79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 1 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/Builder/FrontendDialectTransformer.cpp
View File

@ -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<mlir::Value> 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<mlir::ONNXReshapeOp>(
node, inputs, expectedNumOperands, expectedNumResults);
}
/*!
* Special handle for MaxPool operations.
*/

2
src/Builder/OpBuildTable.inc
View File

@ -223,7 +223,7 @@ if (opName == "ReduceSumSquare")
if (opName == "Relu")
buildOperation<mlir::ONNXReluOp>(node);
if (opName == "Reshape")
ImportNodeReshape(node);
buildOperation<mlir::ONNXReshapeOp>(node);
if (opName == "Resize")
buildOperation<mlir::ONNXResizeOp>(node);
if (opName == "ReverseSequence")

1
utils/gen_onnx_mlir.py
View File

@ -239,7 +239,6 @@ special_op_handler = dict([
("MaxPool", "ImportNodeMaxPool"),
("BatchNormalization", "ImportNodeBatchNormalization"),
("Pad", "ImportNodePad"),
("Reshape", "ImportNodeReshape"),
#("Transpose", "ImportNodeTranspose")
])