Function SetInputsAllInOne

tools/pplnn.cc:608–656

}

static bool SetInputsAllInOne(const string& input_file, const vector<vector<int64_t>>& input_shapes, Runtime* runtime,
                              vector<string>* input_data) {
    Mmap fm;
    auto status = fm.Init(input_file.c_str(), Mmap::READ);
    if (status != RC_SUCCESS) {
        LOG(ERROR) << "mapping file[" << input_file << "] failed.";
        return false;
    }

    auto data = fm.GetData();
    const string file_name = GetBasename(input_file);
    uint64_t expect_input_size = 0;
    for (uint32_t c = 0; c < runtime->GetInputCount(); ++c) {
        auto t = runtime->GetInputTensor(c);

        if (!input_shapes.empty()) {
            t->GetShape()->Reshape(input_shapes[c]);
        }

        expect_input_size += t->GetShape()->CalcBytesExcludingPadding();
    }
    if (fm.GetSize() < expect_input_size) {
        LOG(ERROR) << "input file[" << file_name << "] size(" << fm.GetSize() << ") is less than expected size("
                   << expect_input_size << ")";
        return false;
    }
    if (fm.GetSize() > expect_input_size) {
        LOG(WARNING) << "input file[" << file_name << "] size(" << fm.GetSize() << ") is bigger than expected size("
                     << expect_input_size << ")";
    }

    for (uint32_t c = 0; c < runtime->GetInputCount(); ++c) {
        auto t = runtime->GetInputTensor(c);

        ppl::nn::TensorShape src_desc = *t->GetShape();
        src_desc.SetDataFormat(DATAFORMAT_NDARRAY);
        auto status = t->ConvertFromHost(data, src_desc);
        if (status != RC_SUCCESS) {
            LOG(ERROR) << "convert tensor[" << t->GetName() << "] content failed: " << GetRetCodeStr(status);
            return false;
        }

        const uint64_t content_size = src_desc.CalcBytesIncludingPadding();
        input_data->emplace_back(string(data, content_size));
        data += content_size;
    }

    return true;
}

static const pair<string, datatype_t> g_str2datatype[] = {
    {"fp64", DATATYPE_FLOAT64}, {"fp32", DATATYPE_FLOAT32}, {"fp16", DATATYPE_FLOAT16}, {"int32", DATATYPE_INT32},