Lines Matching refs:num_input1_dims
140 size_t num_input1_dims, in setup_binary_elementwise_nd_f32() argument
161 if (max(num_input1_dims, num_input2_dims) > XNN_MAX_TENSOR_DIMS) { in setup_binary_elementwise_nd_f32()
165 num_input1_dims, num_input2_dims, XNN_MAX_TENSOR_DIMS); in setup_binary_elementwise_nd_f32()
169 for (size_t i = 0; i < num_input1_dims; i++) { in setup_binary_elementwise_nd_f32()
195 const size_t num_common_dims = min(num_input1_dims, num_input2_dims); in setup_binary_elementwise_nd_f32()
197 const size_t input1_dim = input1_shape[num_input1_dims - i]; 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()
237 if (num_input1_dims > num_input2_dims) { in setup_binary_elementwise_nd_f32()
241 for (size_t i = 0; i < num_input1_dims - num_input2_dims; i++) { in setup_binary_elementwise_nd_f32()
246 } else if (num_input2_dims > num_input1_dims) { in setup_binary_elementwise_nd_f32()
250 for (size_t i = 0; i < num_input2_dims - num_input1_dims; i++) { in setup_binary_elementwise_nd_f32()
308 size_t num_input1_dims, in xnn_setup_add_nd_f32() argument
319 num_input1_dims, input1_shape, in xnn_setup_add_nd_f32()
328 size_t num_input1_dims, in xnn_setup_divide_nd_f32() argument
339 num_input1_dims, input1_shape, in xnn_setup_divide_nd_f32()
348 size_t num_input1_dims, in xnn_setup_maximum_nd_f32() argument
359 num_input1_dims, input1_shape, in xnn_setup_maximum_nd_f32()
368 size_t num_input1_dims, in xnn_setup_minimum_nd_f32() argument
379 num_input1_dims, input1_shape, in xnn_setup_minimum_nd_f32()
388 size_t num_input1_dims, in xnn_setup_multiply_nd_f32() argument
399 num_input1_dims, input1_shape, in xnn_setup_multiply_nd_f32()
408 size_t num_input1_dims, in xnn_setup_subtract_nd_f32() argument
419 num_input1_dims, input1_shape, in xnn_setup_subtract_nd_f32()