Kompute
|
Public Member Functions | |
TensorT (std::shared_ptr< vk::PhysicalDevice > physicalDevice, std::shared_ptr< vk::Device > device, const std::vector< T > &data, const TensorTypes &tensorType=TensorTypes::eDevice) | |
T * | data () |
std::vector< T > | vector () |
T & | operator[] (int index) |
void | setData (const std::vector< T > &data) |
TensorDataTypes | dataType () |
![]() | |
Tensor (std::shared_ptr< vk::PhysicalDevice > physicalDevice, std::shared_ptr< vk::Device > device, void *data, uint32_t elementTotalCount, uint32_t elementMemorySize, const TensorDataTypes &dataType, const TensorTypes &tensorType=TensorTypes::eDevice) | |
virtual | ~Tensor () |
void | rebuild (void *data, uint32_t elementTotalCount, uint32_t elementMemorySize) |
void | destroy () |
bool | isInit () |
TensorTypes | tensorType () |
void | recordCopyFrom (const vk::CommandBuffer &commandBuffer, std::shared_ptr< Tensor > copyFromTensor) |
void | recordCopyFromStagingToDevice (const vk::CommandBuffer &commandBuffer) |
void | recordCopyFromDeviceToStaging (const vk::CommandBuffer &commandBuffer) |
void | recordPrimaryBufferMemoryBarrier (const vk::CommandBuffer &commandBuffer, vk::AccessFlagBits srcAccessMask, vk::AccessFlagBits dstAccessMask, vk::PipelineStageFlagBits srcStageMask, vk::PipelineStageFlagBits dstStageMask) |
void | recordStagingBufferMemoryBarrier (const vk::CommandBuffer &commandBuffer, vk::AccessFlagBits srcAccessMask, vk::AccessFlagBits dstAccessMask, vk::PipelineStageFlagBits srcStageMask, vk::PipelineStageFlagBits dstStageMask) |
vk::DescriptorBufferInfo | constructDescriptorBufferInfo () |
uint32_t | size () |
uint32_t | dataTypeMemorySize () |
uint32_t | memorySize () |
TensorDataTypes | dataType () |
void * | rawData () |
void | setRawData (const void *data) |
template<typename T > | |
T * | data () |
template<typename T > | |
std::vector< T > | vector () |
Additional Inherited Members | |
![]() | |
enum | TensorTypes { TensorTypes::eDevice = 0, TensorTypes::eHost = 1, TensorTypes::eStorage = 2 } |
enum | TensorDataTypes { eBool = 0, eInt = 1, eUnsignedInt = 2, eFloat = 3, eDouble = 4 } |
![]() | |
TensorTypes | mTensorType |
TensorDataTypes | mDataType |
uint32_t | mSize |
uint32_t | mDataTypeMemorySize |
void * | mRawData |