Function arm_nn_broadcast_dims_valid¶

Defined in File arm_nn_broadcast_walk.h

Function Documentation¶

static inline int32_t arm_nn_broadcast_dims_valid(const cmsis_nn_dims *dims_1, const cmsis_nn_dims *dims_2, const cmsis_nn_dims *dims_out)¶

Check that two NHWC operands broadcast to the given output shape.

Every kernel that uses ARM_NN_BROADCAST_WALK_NHWC must reject arguments that fail this check, since the walk indexes each input by its own dims and writes the output by the output dims.