Function SetRandomInputs

samples/cpp/others/pplnn_multi_numa_node.cc:261–305

}

static bool SetRandomInputs(const vector<vector<int64_t>>& input_shapes, Runtime* runtime, vector<string>* input_data) {
    for (uint32_t c = 0; c < runtime->GetInputCount(); ++c) {
        auto t = runtime->GetInputTensor(c);
        auto shape = t->GetShape();

        if (input_shapes.empty()) {
            auto dim_count = shape->GetRealDimCount();
            auto dims = GenerateRandomDims(dim_count);
            for (uint32_t j = 2; j < dim_count; ++j) {
                if (shape->GetDim(j) == 1) {
                    shape->SetDim(j, dims[j]);
                }
            }
        } else {
            shape->Reshape(input_shapes[c]);
        }

        auto nr_element = shape->CalcBytesIncludingPadding() / sizeof(float);
        vector<float> buffer(nr_element);

        std::default_random_engine eng;
        std::uniform_real_distribution<float> dis(-1.0f, 1.0f);
        for (uint32_t i = 0; i < nr_element; ++i) {
            buffer[i] = dis(eng);
        }

        auto status = t->ReallocBuffer();
        if (status != RC_SUCCESS) {
            LOG(ERROR) << "ReallocBuffer for tensor[" << t->GetName() << "] failed: " << GetRetCodeStr(status);
            return false;
        }

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

        input_data->emplace_back(string((const char*)buffer.data(), buffer.size() * sizeof(float)));
    }

    return true;
}

static bool SetInputsAllInOne(const string& input_file, const vector<vector<int64_t>>& input_shapes, Runtime* runtime,
                              vector<string>* input_data) {