TensorBuffer的构造函数,用来创建TensorBuffer对象,根据传入的不同参数可选用不同的构造函数。
1 | TensorBuffer::TensorBuffer(); |
1 | TensorBuffer::TensorBuffer(uint32_t size, MemoryData::MemoryType type, int32_t deviceId); |
1 | TensorBuffer::TensorBuffer(uint32_t size, int32_t deviceId); |
1 | TensorBuffer::TensorBuffer(uint32_t size); |
1 | TensorBuffer::TensorBuffer(void *ptr, uint32_t size, MemoryData::MemoryType type = MemoryData::MemoryType::MEMORY_HOST_NEW, int32_t deviceId = -1); |
参数名 |
输入/输出 |
说明 |
---|---|---|
ptr |
输入 |
内存指针。 |
size |
输入 |
内存大小。 |
type |
输入 |
用于构造TensorBuffer对象的内存类型。 |
deviceId |
输入 |
用于构造TensorBuffer对象的参数,int类型数据,设备编号。 |