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