下载
中文
注册

GetBufferCopyType

函数功能

获取两块TensorBuffer内存的数据类型。

函数原型

static TensorBufferCopyType GetBufferCopyType(const TensorBuffer &buffer1, const TensorBuffer &buffer2);

参数说明

参数名

输入/输出

说明

buffer1

输入

TensorBuffer对象。

buffer2

输入

TensorBuffer对象。

返回参数说明

数据结构

说明

TensorBufferCopyType

两块内存拷贝类型。

  • HOST_AND_HOST:两块都为Host侧内存。
  • HOST_AND_DEVICE:一块为Host侧内存,另一块为Device侧。
  • DEVICE_AND_SAME_DEVICE:两块都为Device侧内存,且deviceId相同。
  • DEVICE_AND_DIFF_DEVICE:两块都为Device侧内存,且deviceId不同。