Browse Source

add force infershape for some op

tags/v1.2.0
wxl 3 years ago
parent
commit
365401b52f
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc
  2. +1
    -0
      ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc

+ 1
- 0
ge/ge_local_engine/ops_kernel_store/op/ge_deleted_op.cc View File

@@ -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);


+ 1
- 0
ge/hybrid/node_executor/ge_local/ge_local_node_executor.cc View File

@@ -33,6 +33,7 @@ const std::map<std::string, std::vector<uint32_t>>
{RESHAPE, {}},
{EXPANDDIMS, {}},
{SQUEEZE, {}},
{UNSQUEEZE, {}},
{BROADCASTGRADIENTARGS, {}}
};



Loading…
Cancel
Save