Searched refs:num_input1_dims (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/src/operators/ |
| D | binary-elementwise-nd.c | 375 size_t num_input1_dims, in setup_binary_elementwise_nd() argument
412 if (max(num_input1_dims, num_input2_dims) > XNN_MAX_TENSOR_DIMS) { in setup_binary_elementwise_nd()
416 …xnn_operator_type_to_string(binary_elementwise_op->type), num_input1_dims, num_input2_dims, XNN_MA… in setup_binary_elementwise_nd()
420 for (size_t i = 0; i < num_input1_dims; i++) { in setup_binary_elementwise_nd()
450 const size_t num_common_dims = min(num_input1_dims, num_input2_dims); in setup_binary_elementwise_nd()
452 const size_t input1_dim = input1_shape[num_input1_dims - i]; in setup_binary_elementwise_nd()
489 num_input1_dims - i, input1_dim, num_input2_dims - i, input2_dim); in setup_binary_elementwise_nd()
494 if (num_input1_dims > num_input2_dims) { in setup_binary_elementwise_nd()
498 for (size_t i = 0; i < num_input1_dims - num_input2_dims; i++) { in setup_binary_elementwise_nd()
503 } else if (num_input2_dims > num_input1_dims) { in setup_binary_elementwise_nd()
[all …]
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 51 return i < num_input1_dims() ? this->input1_shape_[i] : 1; in input1_dim()
54 inline size_t num_input1_dims() const { in num_input1_dims() function
221 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestQS8()
307 num_input1_dims(), in TestQS8()
359 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF16()
451 num_input1_dims(), in TestF16()
462 num_input1_dims(), in TestF16()
510 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF32()
628 num_input1_dims(), in TestF32()
639 num_input1_dims(), in TestF32()
[all …]
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1124 size_t num_input1_dims,
1300 size_t num_input1_dims,
1437 size_t num_input1_dims,
1452 size_t num_input1_dims,
1469 size_t num_input1_dims,
1605 size_t num_input1_dims,
1622 size_t num_input1_dims,
1813 size_t num_input1_dims,
1893 size_t num_input1_dims,
1920 size_t num_input1_dims,
|