|
@@ -64,6 +64,7 @@ class UtestGraphPassesFoldingKernelFillKernel : public testing::Test { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -124,6 +125,7 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillBoolShape2And3) { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -230,6 +232,7 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillDimsHaveNegativeNumber) { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -284,6 +287,7 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillDimsTypeNotSupport) { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -310,6 +314,7 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillDimsOverflow) { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -336,6 +341,7 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillDimsMulDataTypeOverflow) { |
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(dims_tensor_desc); |
|
|
|
|
|
|
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
std::vector<GeTensorPtr> outputs; |
|
|
std::vector<GeTensorPtr> outputs; |
|
@@ -343,3 +349,33 @@ TEST_F(UtestGraphPassesFoldingKernelFillKernel, FillDimsMulDataTypeOverflow) { |
|
|
|
|
|
|
|
|
EXPECT_EQ(PARAM_INVALID, status); |
|
|
EXPECT_EQ(PARAM_INVALID, status); |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
TEST_F(UtestGraphPassesFoldingKernelFillKernel, OutputdescUnknown) { |
|
|
|
|
|
ge::OpDescPtr op_dims = std::make_shared<ge::OpDesc>(); |
|
|
|
|
|
vector <int64_t> dims_vec = {2}; |
|
|
|
|
|
vector <int32_t> dims_value_vec = {2, 3}; |
|
|
|
|
|
GeTensorDesc dims_tensor_desc(GeShape(dims_vec), FORMAT_NCHW, DT_INT32); |
|
|
|
|
|
GeTensorPtr dim_tensor = std::make_shared<GeTensor>(dims_tensor_desc, (uint8_t *) dims_value_vec.data(), |
|
|
|
|
|
dims_value_vec.size() * sizeof(int32_t)); |
|
|
|
|
|
OpDescUtils::SetWeights(op_dims, dim_tensor); |
|
|
|
|
|
|
|
|
|
|
|
ge::OpDescPtr op_value = std::make_shared<ge::OpDesc>(); |
|
|
|
|
|
vector <uint8_t> data_vec = {1}; |
|
|
|
|
|
GeTensorDesc value_tensor_desc(GeShape(), FORMAT_NCHW, DT_BOOL); |
|
|
|
|
|
GeTensorPtr value_tensor = |
|
|
|
|
|
std::make_shared<GeTensor>(value_tensor_desc, (uint8_t *) data_vec.data(), data_vec.size() * sizeof(bool)); |
|
|
|
|
|
OpDescUtils::SetWeights(op_value, value_tensor); |
|
|
|
|
|
|
|
|
|
|
|
op_desc_ptr->AddInputDesc(dims_tensor_desc); |
|
|
|
|
|
op_desc_ptr->AddInputDesc(value_tensor_desc); |
|
|
|
|
|
|
|
|
|
|
|
vector <int64_t> out_vec = {-1, -1}; |
|
|
|
|
|
GeTensorDesc out_tensor_desc(GeShape(out_vec), FORMAT_NCHW, DT_INT32); |
|
|
|
|
|
op_desc_ptr->AddOutputDesc(out_tensor_desc); |
|
|
|
|
|
|
|
|
|
|
|
std::vector <ge::ConstGeTensorPtr> input = {dim_tensor, value_tensor}; |
|
|
|
|
|
std::vector <GeTensorPtr> outputs; |
|
|
|
|
|
Status status = kernel->Compute(op_desc_ptr, input, outputs); |
|
|
|
|
|
|
|
|
|
|
|
EXPECT_EQ(NOT_CHANGED, status); |
|
|
|
|
|
} |