GetBufferCopyType所使用的枚举类型，具体请参见GetBufferCopyType。
1 2 3 4 5 6 | enum TensorBufferCopyType { HOST_AND_HOST = 0, HOST_AND_DEVICE, DEVICE_AND_SAME_DEVICE, DEVICE_AND_DIFF_DEVICE }; |
参数名 |
说明 |
|---|---|
HOST_AND_HOST |
两块都为Host侧内存。 |
HOST_AND_DEVICE |
一块为Host侧内存，另一块为Device侧。 |
DEVICE_AND_SAME_DEVICE |
两块都为Device侧内存，且deviceId相同。 |
DEVICE_AND_DIFF_DEVICE |
两块都为Device侧内存，且deviceId不同。 |