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.cc178 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
188 tflite::testing::TestComparisonBool(tflite::BuiltinOperator_EQUAL, input1_dim, in TF_LITE_MICRO_TEST()
194 int input1_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()
210 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
219 tflite::testing::TestComparisonInt(tflite::BuiltinOperator_EQUAL, input1_dim, in TF_LITE_MICRO_TEST()
225 int input1_dim[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
235 tflite::testing::TestComparisonInt(tflite::BuiltinOperator_EQUAL, input1_dim, in TF_LITE_MICRO_TEST()
241 int input1_dim[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
252 tflite::testing::TestComparisonInt(tflite::BuiltinOperator_EQUAL, input1_dim, in TF_LITE_MICRO_TEST()
[all …]
/external/XNNPACK/src/
Dbinary-elementwise-nd.c197 const size_t input1_dim = input1_shape[num_input1_dims - i]; in setup_binary_elementwise_nd_f32() local
199 if (input1_dim == 1 && input2_dim == 1) { in setup_binary_elementwise_nd_f32()
204 if (input1_dim == 1) { in setup_binary_elementwise_nd_f32()
218 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()
219 compressed_output_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()
220 } else if (input1_dim == input2_dim) { in setup_binary_elementwise_nd_f32()
226 compressed_input1_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()
227 compressed_input2_shape[num_compressed_dims - 1] *= input1_dim; in setup_binary_elementwise_nd_f32()
228 compressed_output_shape[num_compressed_dims - 1] *= input1_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()
[all …]
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h46 inline size_t input1_dim(size_t i) const { in input1_dim() function