IsHost
~TensorBuffer
函数功能
TensorBuffer类的默认析构函数。
函数原型
~TensorBuffer() {};
父主题： TensorBuffer
IsDevice
函数功能
用于判断TensorBuffer对象是否在Device侧。
函数原型
bool IsDevice() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| bool | - true：在Device侧。 - false：不在Device侧。 |
父主题： TensorBuffer
IsHost
函数功能
用于判断TensorBuffer对象是否在Host侧。
函数原型
bool IsHost() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| bool | - true：在Host侧。 - false：不在Host侧。 |
父主题： TensorBuffer
SetContext
函数功能
根据TensorBuffer对象中的deviceId，设置线程的Context。
函数原型
APP_ERROR SetContext() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
TensorBufferMalloc
函数功能
用于获取TensorBuffer对象的内存。
Tensor对象在析构时会自动释放内存，因此无需通过函数释放内存。
函数原型
static APP_ERROR TensorBufferMalloc(TensorBuffer &buffer);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| buffer | 输入 | 用来获取对应内存的TensorBuffer数据。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
TensorBufferCopy
函数功能
内存复制函数，根据src中指定的内存位置在Host侧和Device侧之间进行复制。
函数原型
static APP_ERROR TensorBufferCopy(TensorBuffer &dst, const TensorBuffer &src);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| dst | 输出 | 复制后的目标内存。 |
| src | 输入 | 待复制的源内存。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBuffer
在线提单