Function arm_nn_broadcast_dim_valid

Function Documentation

static inline int32_t arm_nn_broadcast_dim_valid(const int32_t dim_1, const int32_t dim_2, const int32_t dim_out)

Check that one NHWC dimension of two operands broadcasts to the output dimension.

TensorFlow Lite broadcast rules: both inputs must be at least 1 (an empty tensor is rejected rather than treated as a no-op), they must be equal or one of them must be 1, and the output dimension must be the larger of the two.