Searched refs:perm_count (Results 1 – 13 of 13) sorted by relevance
28 params.perm_count = 1; in TEST()37 EXPECT_EQ(params.perm_count, 1); in TEST()46 params.perm_count = 2; in TEST()56 EXPECT_EQ(params.perm_count, 2); in TEST()66 params.perm_count = 2; in TEST()76 EXPECT_EQ(params.perm_count, 1); in TEST()85 params.perm_count = 3; in TEST()96 EXPECT_EQ(params.perm_count, 3); in TEST()107 params.perm_count = 3; in TEST()119 EXPECT_EQ(params.perm_count, 2); in TEST()[all …]
57 TFLITE_DCHECK_EQ(params->perm_count, dims_cnt); in RemoveOneSizeDimensions()76 params->perm_count = 1; in RemoveOneSizeDimensions()104 new_params.perm_count = new_dims_cnt; in RemoveOneSizeDimensions()128 for (int i = 0; i < params.perm_count; ++i) { in Flatten()138 const int new_dims_cnt = params.perm_count - skip_dims_cnt; in Flatten()141 non_flatten_params->perm_count = new_dims_cnt; in Flatten()143 for (int i = skip_dims_cnt; i < params.perm_count; ++i) { in Flatten()
1073 int8 perm_count; member
77 params.perm_count = perms.size(); in ComputeExpectedTransposeResult()
102 params.perm_count = size; in Eval()
51 params.perm_count = perms.size(); in RunTestPermutation()
212 transpose_params.perm_count = 4; in ResizeAndTransposeWeights()
74 transpose_params.perm_count = 4; in PopulateSubGraph()
184 transpose_params.perm_count = 4; in PopulateSubGraph()
1000 params.perm_count = 4; in Transpose()
2309 TFLITE_DCHECK_EQ(unextended_output_size, params.perm_count); in TransposeImpl()
7659 TFLITE_DCHECK_EQ(output_size, unshrinked_params.perm_count); in Transpose()7674 for (int i = 0; i < shrinked_params.perm_count; ++i) { in Transpose()
4930 params.perm_count = 4; in Transpose()