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