Fix unified_api_test

This commit is contained in:
Võ Văn Nghĩa 2020-12-10 07:52:26 +07:00
parent c18a4cade5
commit 9701a12b48

View File

@ -119,7 +119,7 @@ TEST_P(UnifiedAPI, TestTensorShape2x4) {
{
AbstractTensorHandle* x_raw = nullptr;
float data[] = {0., 0., 0., 0., 0., 0., 0., 0};
int64 dim_sizes[] = {2, 4};
int64_t dim_sizes[] = {2, 4};
Status s =
TestTensorHandleWithDimsFloat(ctx.get(), data, dim_sizes, 2, &x_raw);
ASSERT_EQ(errors::OK, s.code()) << s.error_message();