Searched refs:input2_dim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | comparisons_test.cc | 161 int input2_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()
177 int input2_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()
193 int input2_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()
208 int input2_dim[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
218 tflite::ops::micro::Register_EQUAL(), input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
224 int input2_dim[] = {4, 1, 1, 1, 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/ |
| D | binary-elementwise-nd.c | 453 const size_t input2_dim = input2_shape[num_input2_dims - i]; in setup_binary_elementwise_nd() local
454 if (input1_dim == 1 && input2_dim == 1) { in setup_binary_elementwise_nd()
465 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()
466 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()
467 } else if (input2_dim == 1) { in setup_binary_elementwise_nd()
475 } else if (input1_dim == input2_dim) { in setup_binary_elementwise_nd()
489 num_input1_dims - i, input1_dim, num_input2_dims - i, input2_dim); in setup_binary_elementwise_nd()
508 const size_t input2_dim = input2_shape[i]; in setup_binary_elementwise_nd() local
509 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()
510 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd()
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 92 inline size_t input2_dim(size_t i) const { in input2_dim() function
|