Function checked_product

src/framework/runtime/host_ops.cpp:13–19

namespace {

int64_t checked_product(std::initializer_list<int64_t> dims) {
    int64_t result = 1;
    for (const int64_t dim : dims) {
        result *= dim;
    }
    return result;
}

float stable_softmax_pick(const float * row, int64_t vocab_size, float uniform) {
    float max_value = row[0];