|
|
|
@@ -361,6 +361,9 @@ TEST_F(UtestGeHybrid, TestTaskContext) { |
|
|
|
desc = task_context->MutableOutputDesc(0); |
|
|
|
ASSERT_TRUE(desc != nullptr); |
|
|
|
ASSERT_EQ(desc->GetShape().GetDims(), shape.GetDims()); |
|
|
|
GeTensorDesc output_desc; |
|
|
|
ASSERT_EQ(task_context->GetOutputDesc(0, output_desc), SUCCESS); |
|
|
|
ASSERT_EQ(output_desc.GetShape().GetDims(), shape.GetDims()); |
|
|
|
|
|
|
|
desc = task_context->MutableInputDesc(0); |
|
|
|
ASSERT_TRUE(desc != nullptr); |
|
|
|
|