Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons_test.cc179 int input2_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
189 input1_data, input2_dim, input2_data, in TF_LITE_MICRO_TEST()
195 int input2_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
205 tflite::BuiltinOperator_EQUAL, input1_dim, input1_data, input2_dim, in TF_LITE_MICRO_TEST()
211 int input2_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
220 input1_data, input2_dim, input2_data, in TF_LITE_MICRO_TEST()
226 int input2_dim[] = {4, 1, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
236 input1_data, input2_dim, input2_data, in TF_LITE_MICRO_TEST()
242 int input2_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
253 input1_data, input2_dim, input2_data, in TF_LITE_MICRO_TEST()
[all …]
/external/XNNPACK/src/
Dbinary-elementwise-nd.c198 const size_t input2_dim = input2_shape[num_input2_dims - i]; in setup_binary_elementwise_nd_f32() local
199 if (input1_dim == 1 && input2_dim == 1) { in setup_binary_elementwise_nd_f32()
210 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()
211 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()
212 } else if (input2_dim == 1) { in setup_binary_elementwise_nd_f32()
220 } else if (input1_dim == input2_dim) { in setup_binary_elementwise_nd_f32()
232 num_input1_dims - i, input1_dim, num_input2_dims - i, input2_dim); in setup_binary_elementwise_nd_f32()
251 const size_t input2_dim = input2_shape[i]; in setup_binary_elementwise_nd_f32() local
252 compressed_input2_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()
253 compressed_output_shape[num_compressed_dims - 1] *= input2_dim; in setup_binary_elementwise_nd_f32()
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h69 inline size_t input2_dim(size_t i) const { in input2_dim() function