CheckAndMoveTensors
函数功能
保护成员函数，调用IsValidTensors校验张量无误后将张量内存搬运至Host侧。
函数原型
APP_ERROR PostProcessBase::CheckAndMoveTensors(std::vector<TensorBase> &tensors);
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
tensors |
输入 |
输入张量。 |
返回参数说明
数据结构 |
说明 |
|---|---|
APP_ERROR |
程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： PostProcessBase