From 365401b52fe53306f7b3ef87e4a2b17ac8090911 Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Tue, 9 Mar 2021 19:57:27 +0800
Subject: [PATCH] add force infershape for some op

---
 ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc    | 1 +
 ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc b/ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc
index b2f3d095..90d95217 100755
--- a/ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc
+++ b/ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc
@@ -38,6 +38,7 @@ REGISTER_OP_CREATOR(ExpandDims, GeDeletedOp);
 REGISTER_OP_CREATOR(Reshape, GeDeletedOp);
 REGISTER_OP_CREATOR(ReFormat, GeDeletedOp);
 REGISTER_OP_CREATOR(Squeeze, GeDeletedOp);
+REGISTER_OP_CREATOR(Unsqueeze, GeDeletedOp);
 REGISTER_OP_CREATOR(Size, GeDeletedOp);
 REGISTER_OP_CREATOR(Shape, GeDeletedOp);
 REGISTER_OP_CREATOR(ShapeN, GeDeletedOp);
diff --git a/ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc b/ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc
index 3d2e3084..9d92420e 100755
--- a/ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc
+++ b/ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc
@@ -33,6 +33,7 @@ const std::map<std::string, std::vector<uint32_t>>
                                           {RESHAPE, {}},
                                           {EXPANDDIMS, {}},
                                           {SQUEEZE, {}},
+                                          {UNSQUEEZE, {}},
                                           {BROADCASTGRADIENTARGS, {}}
                                          };