|
|
@@ -230,7 +230,7 @@ TEST_F(UtestGeTensor, test_tensor_invalid_null) { |
|
|
|
GeTensor tensor(msg_owner, nullptr); |
|
|
|
EXPECT_EQ(tensor.GetData().size(), 0); |
|
|
|
EXPECT_EQ(tensor.MutableData().size(), 0); |
|
|
|
EXPECT_EQ(tensor.SetData(Buffer(100)), ge::GRAPH_PARAM_INVALID); |
|
|
|
EXPECT_EQ(tensor.SetData(Buffer(100)), GRAPH_SUCCESS); |
|
|
|
|
|
|
|
TensorUtils::SetWeightSize(tensor.MutableTensorDesc(), 100); |
|
|
|
EXPECT_EQ(TensorUtils::GetWeightSize(tensor), 0); |
|
|
|