|
|
@@ -31,6 +31,7 @@ class UtestGeGenerator : public testing::Test { |
|
|
|
void TearDown() {} |
|
|
|
}; |
|
|
|
|
|
|
|
/* |
|
|
|
TEST_F(UtestGeGenerator, test_build_single_op_offline) { |
|
|
|
GeTensorDesc tensor_desc(GeShape(), FORMAT_NCHW, DT_FLOAT); |
|
|
|
TensorUtils::SetSize(tensor_desc, 512); |
|
|
@@ -52,6 +53,7 @@ TEST_F(UtestGeGenerator, test_build_single_op_offline) { |
|
|
|
generator.Initialize({}); |
|
|
|
EXPECT_EQ(generator.BuildSingleOpModel(op_desc, inputs, outputs, "offline_"), GE_GENERATOR_GRAPH_MANAGER_BUILD_GRAPH_FAILED); |
|
|
|
} |
|
|
|
*/ |
|
|
|
|
|
|
|
TEST_F(UtestGeGenerator, test_build_single_op_online) { |
|
|
|
GeTensorDesc tensor_desc; |
|
|
|