Searched refs:num_input1_dims (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/src/ |
| D | binary-elementwise-nd.c | 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()
[all …]
|
| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 47 return i < num_input1_dims() ? this->input1_shape_[i] : 1; in input1_dim()
50 inline size_t num_input1_dims() const { in num_input1_dims() function
151 … std::copy(input1_shape().cbegin(), input1_shape().cend(), input1_dims.end() - num_input1_dims()); in TestF32()
264 num_input1_dims(), in TestF32()
275 num_input1_dims(), in TestF32()
286 num_input1_dims(), in TestF32()
297 num_input1_dims(), in TestF32()
308 num_input1_dims(), in TestF32()
319 num_input1_dims(), in TestF32()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 515 size_t num_input1_dims,
665 size_t num_input1_dims,
758 size_t num_input1_dims,
773 size_t num_input1_dims,
790 size_t num_input1_dims,
870 size_t num_input1_dims,
|