@@ -57,7 +57,7 @@ TEST(TFUtilsTest, LoadAndExecuteTest) {
57
57
auto ER = Evaluator.evaluate ();
58
58
EXPECT_TRUE (ER.hasValue ());
59
59
float Ret = *ER->getTensorValue <float >(0 );
60
- EXPECT_EQ (static_cast <size_t >(Ret), 80 );
60
+ EXPECT_EQ (static_cast <int64_t >(Ret), 80 );
61
61
EXPECT_EQ (ER->getUntypedTensorValue (0 ),
62
62
reinterpret_cast <const void *>(ER->getTensorValue <float >(0 )));
63
63
}
@@ -72,7 +72,7 @@ TEST(TFUtilsTest, LoadAndExecuteTest) {
72
72
auto ER = Evaluator.evaluate ();
73
73
EXPECT_TRUE (ER.hasValue ());
74
74
float Ret = *ER->getTensorValue <float >(0 );
75
- EXPECT_EQ (static_cast <size_t >(Ret), 80 );
75
+ EXPECT_EQ (static_cast <int64_t >(Ret), 80 );
76
76
}
77
77
}
78
78
@@ -135,10 +135,10 @@ TEST(TFUtilsTest, TensorSpecSizesAndTypes) {
135
135
auto Spec3DLarge = TensorSpec::createSpec<float >(" Hi3" , {2 , 4 , 10 });
136
136
EXPECT_TRUE (Spec1D.isElementType <int16_t >());
137
137
EXPECT_FALSE (Spec3DLarge.isElementType <double >());
138
- EXPECT_EQ (Spec1D.getElementCount (), 1 );
139
- EXPECT_EQ (Spec2D.getElementCount (), 1 );
140
- EXPECT_EQ (Spec1DLarge.getElementCount (), 10 );
141
- EXPECT_EQ (Spec3DLarge.getElementCount (), 80 );
138
+ EXPECT_EQ (Spec1D.getElementCount (), 1U );
139
+ EXPECT_EQ (Spec2D.getElementCount (), 1U );
140
+ EXPECT_EQ (Spec1DLarge.getElementCount (), 10U );
141
+ EXPECT_EQ (Spec3DLarge.getElementCount (), 80U );
142
142
EXPECT_EQ (Spec3DLarge.getElementByteSize (), sizeof (float ));
143
143
EXPECT_EQ (Spec1D.getElementByteSize (), sizeof (int16_t ));
144
144
}
0 commit comments