Home
last modified time | relevance | path

Searched refs:input1_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc160 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
171 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
176 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
187 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
192 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
202 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
207 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
218 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
223 int input1_dim[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
235 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
[all …]
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c452 const size_t input1_dim = input1_shape[num_input1_dims - i]; in setup_binary_elementwise_nd() local
454 if (input1_dim == 1 && input2_dim == 1) { in setup_binary_elementwise_nd()
459 if (input1_dim == 1) { in setup_binary_elementwise_nd()
473 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()
474 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()
475 } else if (input1_dim == input2_dim) { in setup_binary_elementwise_nd()
481 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()
482 compressed_input2_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()
483 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd()
489 num_input1_dims - i, input1_dim, num_input2_dims - i, input2_dim); in setup_binary_elementwise_nd()
[all …]
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h50 inline size_t input1_dim(size_t i) const { in input1_dim() function