saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cosmetic change in non tensor tests (#5317)

Browse source
This commit is contained in:
Hariharan Seshadri 2020-09-28 21:23:30 -07:00 committed by GitHub
parent 1ff3b2d5b8
commit cb83097632
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
onnxruntime/test/shared_lib/test_nontensor_types.cc
View file

@ -179,8 +179,10 @@ TEST(CApiTest, TypeInfoMap) {
#if !defined(DISABLE_ML_OPS)
Ort::Value map_ort = Ort::Value::CreateMap(keys_tensor, values_tensor);
Ort::TypeInfo type_info = map_ort.GetTypeInfo();
//It doesn't own the pointer
Ort::MapTypeInfo map_type_info = type_info.GetMapTypeInfo();
//It doesn't own the pointer -
//The destructor of the "Unowned" struct will release the ownership (and thus prevent the pointer from being double freed)
auto map_type_info = type_info.GetMapTypeInfo();
//Check key type
ASSERT_EQ(map_type_info.GetMapKeyType(), ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64);
@ -194,8 +196,6 @@ TEST(CApiTest, TypeInfoMap) {
// ASSERT_EQ(map_value_type_info.GetTensorTypeAndShapeInfo().GetShape(), dims);
ASSERT_EQ(map_value_type_info.GetTensorTypeAndShapeInfo().GetElementType(), ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT);
//As it doesn't own the pointer, we should release the ownership, otherwise the pointer will be double freed.
map_type_info.release();
#else
#if !defined(ORT_NO_EXCEPTIONS)
@ -296,13 +296,14 @@ TEST(CApiTest, TypeInfoSequence) {
Ort::Value seq_ort = Ort::Value::CreateSequence(in);
Ort::TypeInfo type_info = seq_ort.GetTypeInfo();
Ort::SequenceTypeInfo seq_type_info = type_info.GetSequenceTypeInfo();
//It doesn't own the pointer -
//The destructor of the "Unowned" struct will release the ownership (and thus prevent the pointer from being double freed)
auto seq_type_info = type_info.GetSequenceTypeInfo();
ASSERT_EQ(seq_type_info.GetSequenceElementType().GetONNXType(), ONNX_TYPE_TENSOR);
// No shape present, as sequence allows different shapes for each element
// ASSERT_EQ(seq_type_info.GetSequenceElementType().GetTensorTypeAndShapeInfo().GetShape(), dims);
ASSERT_EQ(seq_type_info.GetSequenceElementType().GetTensorTypeAndShapeInfo().GetElementType(),
ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64);
seq_type_info.release();
}