Searched refs:non_flatten_params (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 287 TransposeParams non_flatten_params; in TEST() local
290 &non_flatten_output_shape, &non_flatten_params); in TEST()
296 EXPECT_EQ(non_flatten_params.perm_count, 2); in TEST()
297 EXPECT_EQ(non_flatten_params.perm[0], 1); in TEST()
298 EXPECT_EQ(non_flatten_params.perm[1], 0); in TEST()
314 TransposeParams non_flatten_params; in TEST() local
317 &non_flatten_output_shape, &non_flatten_params); in TEST()
323 EXPECT_EQ(non_flatten_params.perm_count, 3); in TEST()
324 EXPECT_EQ(non_flatten_params.perm[0], 1); in TEST()
325 EXPECT_EQ(non_flatten_params.perm[1], 0); in TEST()
[all …]
|
| D | transpose_utils.cc | 124 TransposeParams* non_flatten_params) { in Flatten() argument
141 non_flatten_params->perm_count = new_dims_cnt; in Flatten()
146 non_flatten_params->perm[i - skip_dims_cnt] = params.perm[i]; in Flatten()
151 if (non_flatten_params->perm[j] >= i && in Flatten()
152 (min_val_idx == -1 || non_flatten_params->perm[min_val_idx] > in Flatten()
153 non_flatten_params->perm[j])) { in Flatten()
157 non_flatten_params->perm[min_val_idx] = i; in Flatten()
|
| D | transpose_utils.h | 46 TransposeParams* non_flatten_params);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7898 TransposeParams non_flatten_params; in Transpose() local
7903 &non_flatten_params); in Transpose()
7904 TFLITE_DCHECK_NE(non_flatten_params.perm[0], 0); in Transpose()
7907 TransposeImpl<T, N>(non_flatten_params, non_flatten_input_shape, in Transpose()
|