Searched refs:validateShapes (Results 1 – 1 of 1) sorted by relevance
185 bool validateShapes(const Shape& input, const Shape& weights, const Shape& bias, in validateShapes() function288 NN_RET_CHECK(validateShapes(input, weights, bias)); in validate()300 NN_RET_CHECK(validateShapes(input, weights, bias, &output)); in prepare()