Searched refs:XNN_MAX_TENSOR_DIMS (Results 1 – 6 of 6) sorted by relevance
37 assert(input1_shape.size() <= XNN_MAX_TENSOR_DIMS); in input1_shape()60 assert(input2_shape.size() <= XNN_MAX_TENSOR_DIMS); in input2_shape()145 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF32()146 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF32()147 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestF32()153 for (size_t i = 0; i < XNN_MAX_TENSOR_DIMS; i++) { in TestF32()163 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_strides; in TestF32()164 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_strides; in TestF32()165 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF32()167 for (size_t i = XNN_MAX_TENSOR_DIMS; i != 0; i--) { in TestF32()
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()184 size_t compressed_input1_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd_f32()185 size_t compressed_input2_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd_f32()186 size_t compressed_output_shape[XNN_MAX_TENSOR_DIMS]; in setup_binary_elementwise_nd_f32()187 for (size_t i = 0; i < XNN_MAX_TENSOR_DIMS; i++) { in setup_binary_elementwise_nd_f32()281 …binary_elementwise_op->context.elementwise_binary.a_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = a_stride… in setup_binary_elementwise_nd_f32()284 …binary_elementwise_op->context.elementwise_binary.b_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = b_stride… in setup_binary_elementwise_nd_f32()286 …binary_elementwise_op->context.elementwise_binary.y_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = y_stride… in setup_binary_elementwise_nd_f32()
41 if (num_dims > XNN_MAX_TENSOR_DIMS) { in xnn_define_tensor_value()43 XNN_MAX_TENSOR_DIMS); in xnn_define_tensor_value()
576 size_t a_stride[XNN_MAX_TENSOR_DIMS - 1];578 size_t b_stride[XNN_MAX_TENSOR_DIMS - 1];580 size_t y_stride[XNN_MAX_TENSOR_DIMS - 1];
21 size_t dim[XNN_MAX_TENSOR_DIMS];
28 #define XNN_MAX_TENSOR_DIMS 6 macro