EXPECT_EQ (dim[0], 345U);
EXPECT_EQ (dim[1], 123U);
EXPECT_EQ (dim[2], 433U);
- EXPECT_EQ (dim[3], 1U);
dim_str = gst_tensor_get_dimension_string (dim);
EXPECT_TRUE (gst_tensor_dimension_string_is_equal (dim_str, "345:123:433:1"));
EXPECT_EQ (rank, 2U);
EXPECT_EQ (dim[0], 345U);
EXPECT_EQ (dim[1], 123U);
- EXPECT_EQ (dim[2], 1U);
- EXPECT_EQ (dim[3], 1U);
dim_str = gst_tensor_get_dimension_string (dim);
EXPECT_TRUE (gst_tensor_dimension_string_is_equal (dim_str, "345:123:1:1"));
rank = gst_tensor_parse_dimension ("345", dim);
EXPECT_EQ (rank, 1U);
EXPECT_EQ (dim[0], 345U);
- EXPECT_EQ (dim[1], 1U);
- EXPECT_EQ (dim[2], 1U);
- EXPECT_EQ (dim[3], 1U);
dim_str = gst_tensor_get_dimension_string (dim);
EXPECT_TRUE (gst_tensor_dimension_string_is_equal (dim_str, "345:1:1:1"));
gst_tensor_meta_info_parse_header (&meta, map.data);
EXPECT_EQ (meta.type, _NNS_FLOAT32);
EXPECT_EQ (meta.dimension[0], 5U);
- EXPECT_EQ (meta.dimension[1], 1U);
hsize = gst_tensor_meta_info_get_header_size (&meta);
ASSERT_EQ (gst_buffer_get_size (out_buf), data_out_size + hsize);
EXPECT_TRUE (gst_tensor_dimension_string_is_equal (input_dim, "3:224:224:1"));
g_free (input_dim);
- /* Rank should be 3 since input dimension string is not given. */
+ /* Rank should be 4 since input dimension string is not given. */
gchar *input_ranks;
g_object_get (filter, "inputranks", &input_ranks, NULL);
- EXPECT_STREQ (input_ranks, "3");
+ EXPECT_STREQ (input_ranks, "4");
g_free (input_ranks);
gchar *output_dim;
g_object_get (filter, "output", &output_dim, NULL);
- EXPECT_TRUE (gst_tensor_dimension_string_is_equal (output_dim, "1001:1:1:1"));
+ EXPECT_TRUE (gst_tensor_dimension_string_is_equal (output_dim, "1001:1"));
g_free (output_dim);
- /* Rank should be 1 since output dimension string is not given. */
+ /* Rank should be 2 since output dimension string is not given. */
gchar *output_ranks;
g_object_get (filter, "outputranks", &output_ranks, NULL);
- EXPECT_STREQ (output_ranks, "1");
+ EXPECT_STREQ (output_ranks, "2");
g_free (output_ranks);
g_object_unref (filter);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_UINT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 10U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 0);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_UINT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 10U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 10);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_UINT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 10U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 0);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_UINT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 10U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 0);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_INT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 5U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 10);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_INT32);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 2U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.info.info[1].type, _NNS_INT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[0], 2U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[1], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 10);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
EXPECT_TRUE (gst_tensors_config_validate (&g_test_data.tensors_config));
EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_UINT8);
EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 10U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 1U);
- EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
EXPECT_EQ (g_test_data.tensors_config.rate_n, 10);
EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
videotestsrc num-buffers=1 ! \
video/x-raw,format=RGB,width=10,height=10,framerate=0/1 ! videoconvert ! \
tensor_converter ! tensor_filter framework=lua \
- model=\"${MAKE_SAMPLE_TENSORS_SCRIPT}\" ! tensor_sparse_enc ! filesink location=./sample1.sparse" 4 0 0 $PERFORMANCE
+ model=\"${MAKE_SAMPLE_TENSORS_SCRIPT}\" ! \
+ other/tensors,num_tensors=1,framerate=0/1,dimensions=1:3:4:1,types=uint8 ! \
+ tensor_sparse_enc ! filesink location=./sample1.sparse" 4 0 0 $PERFORMANCE
gstTest "--gst-plugin-path=${PATH_TO_PLUGIN} \
filesrc location=sample1.sparse ! \
*/
TEST (nnstreamerNnfwRuntimeRawFunctions, getDimension)
{
- int ret, i;
+ int ret;
void *data = NULL;
GstTensorsInfo info, res;
gchar *model_file;
info.num_tensors = 1;
info.info[0].type = _NNS_FLOAT32;
- for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++)
- info.info[0].dimension[i] = 1;
+ info.info[0].dimension[0] = 1;
/** get input/output dimension successfully */
ret = sp->getInputDimension (&prop, &data, &res);
EXPECT_EQ (res.num_tensors, info.num_tensors);
EXPECT_EQ (res.info[0].type, info.info[0].type);
- for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++)
- EXPECT_EQ (res.info[0].dimension[i], info.info[0].dimension[i]);
+ EXPECT_EQ (res.info[0].dimension[0], info.info[0].dimension[0]);
ret = sp->getOutputDimension (&prop, &data, &res);
EXPECT_EQ (ret, 0);
EXPECT_EQ (res.num_tensors, info.num_tensors);
EXPECT_EQ (res.info[0].type, info.info[0].type);
- for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++)
- EXPECT_EQ (res.info[0].dimension[i], info.info[0].dimension[i]);
+ EXPECT_EQ (res.info[0].dimension[0], info.info[0].dimension[0]);
sp->close (&prop, &data);
res.num_tensors = 1;
res.info[0].type = _NNS_FLOAT32;
res.info[0].dimension[0] = tensor_size;
- for (i = 1; i < NNS_TENSOR_RANK_LIMIT; i++)
- res.info[0].dimension[i] = 1;
/** get input/output dimension successfully */
ret = sp->getInputDimension (&prop, &data, &in_info);
EXPECT_EQ (res.info[0].type, in_info.info[0].type);
EXPECT_NE (res.info[0].dimension[0], in_info.info[0].dimension[0]);
- for (i = 1; i < NNS_TENSOR_RANK_LIMIT; i++)
- EXPECT_EQ (res.info[0].dimension[i], in_info.info[0].dimension[i]);
ret = sp->setInputDimension (&prop, &data, &res, &out_info);
EXPECT_EQ (ret, 0);
info.info[0].type = _NNS_UINT8;
info.info[0].dimension[0] = 1001;
info.info[0].dimension[1] = 1;
- info.info[0].dimension[2] = 1;
- info.info[0].dimension[3] = 1;
EXPECT_EQ (res.num_tensors, info.num_tensors);
EXPECT_EQ (res.info[0].type, info.info[0].type);
EXPECT_EQ (res.info[0].dimension[0], info.info[0].dimension[0]);
EXPECT_EQ (res.info[0].dimension[1], info.info[0].dimension[1]);
- EXPECT_EQ (res.info[0].dimension[2], info.info[0].dimension[2]);
- EXPECT_EQ (res.info[0].dimension[3], info.info[0].dimension[3]);
-
output.size = gst_tensor_info_get_size (&res.info[0]);
input.data = NULL;