add force infershape for some op

pull/1224/head
wxl 4 years ago
parent c94e0fbdc6
commit 365401b52f

@ -38,6 +38,7 @@ REGISTER_OP_CREATOR(ExpandDims, GeDeletedOp);
REGISTER_OP_CREATOR(Reshape, GeDeletedOp); REGISTER_OP_CREATOR(Reshape, GeDeletedOp);
REGISTER_OP_CREATOR(ReFormat, GeDeletedOp); REGISTER_OP_CREATOR(ReFormat, GeDeletedOp);
REGISTER_OP_CREATOR(Squeeze, GeDeletedOp); REGISTER_OP_CREATOR(Squeeze, GeDeletedOp);
REGISTER_OP_CREATOR(Unsqueeze, GeDeletedOp);
REGISTER_OP_CREATOR(Size, GeDeletedOp); REGISTER_OP_CREATOR(Size, GeDeletedOp);
REGISTER_OP_CREATOR(Shape, GeDeletedOp); REGISTER_OP_CREATOR(Shape, GeDeletedOp);
REGISTER_OP_CREATOR(ShapeN, GeDeletedOp); REGISTER_OP_CREATOR(ShapeN, GeDeletedOp);

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

Loading…
Cancel
Save