4.2.5. IOMode
IOMode用于定义输入Tensor和输出Tensor的内存位置信息(device memory或system memory)。
- 接口形式:
enum IOMode { SYSI, SYSO, SYSIO, DEVIO };
参数说明:
SYSI
输入Tensor在system memory,输出Tensor在device memory
SYSO
输入Tensor在device memory,输出Tensor在system memory
SYSIO
输入Tensor在system memory,输出Tensor在system memory
DEVIO
输入Tensor在device memory,输出Tensor在device memory