跳到主要内容

~TensorContext

CopyBetweenDiffDevice​

函数功能​

将Device侧的src对象拷贝至Device侧的dst对象。两块TensorBuffer内存的deviceId不同。

函数原型​

static APP_ERROR CopyBetweenDiffDevice(TensorBuffer &dst, const TensorBuffer &src);

参数说明​

参数名输入/输出说明
dst输入复制后的目标内存。
src输入待复制的源内存。

返回参数说明​

数据结构说明
APP_ERROR程序执行返回的错误码，请参考APP_ERROR说明。

父主题： TensorBuffer

类说明​

该类主要有TensorContext数据结构的构造方法和相关的功能接口。

父主题： TensorContext

TensorContext​

函数功能​

TensorContext构造函数，创建TensorContext()对象并初始化AscendCL。

函数原型​

TensorContext();

父主题： TensorContext

~TensorContext​

函数功能​

TensorContext类的默认析构函数。

函数原型​

~TensorContext();

父主题： TensorContext

GetInstance​

函数功能​

获取TensorContext单例对象。

函数原型​

static std::shared_ptr<TensorContext> GetInstance();

返回参数说明​

数据结构说明
std::shared_ptrTensorContext单例对象。

父主题： TensorContext

SetContext​

函数功能​

设置线程的Context。

函数原型​

APP_ERROR SetContext(const uint32_t &deviceId);

参数说明​

参数名输入/输出说明
deviceId输入线程当前的deviceId。

返回参数说明​

数据结构说明
APP_ERROR程序执行返回的错误码，请参考APP_ERROR说明。

父主题： TensorContext

类说明​

该类主要有TensorShape数据结构的构造方法和相关的功能接口。

父主题： TensorShape

在线提单