|
|
@@ -410,6 +410,13 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { |
|
|
|
model.model_helper_.model_ = ge_model; |
|
|
|
auto op_desc = std::make_shared<ge::OpDesc>("add", "Add"); |
|
|
|
AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE); |
|
|
|
GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); |
|
|
|
for (int i = 0; i < 4; i++) { |
|
|
|
op_desc->AddInputDesc(tensor); |
|
|
|
} |
|
|
|
for (int i = 0; i < 2; i++) { |
|
|
|
op_desc->AddOutputDesc(tensor); |
|
|
|
} |
|
|
|
NodePtr node = graph->AddNode(op_desc); |
|
|
|
model.op_list_[0] = node; |
|
|
|
StreamResource *res = new (std::nothrow) StreamResource(1); |
|
|
|