Searched refs:num_input2_dims (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/src/ |
| D | binary-elementwise-nd.c | 142 size_t num_input2_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()
176 for (size_t i = 0; i < num_input2_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()
198 const size_t input2_dim = input2_shape[num_input2_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()
[all …]
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 70 return i < num_input2_dims() ? this->input2_shape_[i] : 1; in input2_dim()
73 inline size_t num_input2_dims() const { in num_input2_dims() function
152 … std::copy(input2_shape().cbegin(), input2_shape().cend(), input2_dims.end() - num_input2_dims()); in TestF32()
266 num_input2_dims(), in TestF32()
277 num_input2_dims(), in TestF32()
288 num_input2_dims(), in TestF32()
299 num_input2_dims(), in TestF32()
310 num_input2_dims(), in TestF32()
321 num_input2_dims(), in TestF32()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 517 size_t num_input2_dims,
667 size_t num_input2_dims,
760 size_t num_input2_dims,
775 size_t num_input2_dims,
792 size_t num_input2_dims,
872 size_t num_input2_dims,
|