Searched refs:perm_count (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 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 …]
|
| D | transpose_utils.cc | 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()
|
| D | types.h | 1144 int8_t perm_count; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | transpose.h | 35 TFLITE_DCHECK_EQ(unextended_output_size, params.perm_count); in TransposeImpl()
|
| D | legacy_reference_ops.h | 970 params.perm_count = 4; in Transpose()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose.cc | 102 params.perm_count = size; in Eval()
|
| D | transpose_test.cc | 58 params.perm_count = perms.size(); in RunTestPermutation()
|
| D | transpose_conv.cc | 207 transpose_params.perm_count = 4; in ResizeAndTransposeWeights()
|
| D | batch_matmul.cc | 395 params.perm_count = rank; in TransposeRowsColumnsImpl()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | transpose.cc | 102 params.perm_count = size; in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 47 TransposeParams params = {/*perm_count=*/4, /*perm=*/{0, 3, 1, 2}};
56 TransposeParams params = {/*perm_count=*/4, /*perm=*/{0, 2, 3, 1}};
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_test.cc | 82 params.perm_count = perms.size(); in ComputeExpectedTransposeResult()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 178 transpose_params.perm_count = 4; in InitializeWeightsNodes()
|
| D | matmul_builder.cc | 176 transpose_params.perm_count = 4; in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 1241 int8_t perm_count;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7867 TFLITE_DCHECK_EQ(output_size, unshrinked_params.perm_count); in Transpose()
7882 for (int i = 0; i < shrinked_params.perm_count; ++i) { in Transpose()
|
| D | legacy_optimized_ops.h | 4943 params.perm_count = 4; in Transpose()
|