Home
last modified time | relevance | path

Searched defs:axis_dims (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Darg_min_max_test.cc55 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestArgMinMaxFloat() local
81 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestArgMinMaxQuantized() local
109 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
123 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
138 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
157 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
176 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
195 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
214 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
233 const int axis_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dsplit_test.cc34 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestSplitTwoOutputsFloat() local
91 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestSplitFourOutputsFloat() local
162 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestSplitTwoOutputsQuantized() local
218 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestSplitTwoOutputsQuantized32() local
Dreduce_test.cc98 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestMeanFloatInput4D() local
127 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestReduceOpFloat() local
157 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestReduceOpQuantized() local
195 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestMeanOpQuantized() local
Dsplit_v_test.cc39 TfLiteIntArray* axis_dims = IntArrayFromInts(axis_dims_data); in TestSplitVFloat() local