Function validate_shape

src/framework/core/module.cpp:212–226

}

void validate_shape(const TensorValue & value, const TensorShape & expected, const char * name) {
    if (!value.valid()) {
        throw std::runtime_error(name_or_default(name) + " is invalid");
    }
    if (value.shape.rank != expected.rank) {
        throw std::runtime_error(name_or_default(name) + " rank mismatch: expected " + expected.to_string() +
                                 ", got " + value.shape.to_string());
    }
    for (size_t i = 0; i < expected.rank; ++i) {
        if (value.shape.dims[i] != expected.dims[i]) {
            throw std::runtime_error(name_or_default(name) + " shape mismatch: expected " + expected.to_string() +
                                     ", got " + value.shape.to_string());
        }
    }
}

void validate_last_dim(const TensorValue & value, int64_t expected, const char * name) {
    if (!value.valid()) {