Home
last modified time | relevance | path

Searched refs:m_shapeNum (Results 1 – 1 of 1) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnncompiler.cpp62 ret = tensorDesc.first->GetShape(&m_shape, &m_shapeNum); in CopyFromTensorDesc()
93 ret = tensorDesc.SetShape(m_shape, m_shapeNum); in CopyToTensorDesc()
111 size_t m_shapeNum{0}; member
119 const size_t SIZE_OF_SHAPE_NUM = sizeof(SerializedTensorDesc::m_shapeNum);
803 totalSize += tensorDesc.m_shapeNum * sizeof(int32_t); in SerializeTensorsToBuffer()
849 size_t sizeOfShape = tensorDesc.m_shapeNum * sizeof(int32_t); in SerializeTensorsToBuffer()
914 memRet = memcpy_s(&desc.m_shapeNum, SIZE_OF_SHAPE_NUM, ptr, sizeof(desc.m_shapeNum)); in DeserializedTensorsFromBuffer()
920 ptr += sizeof(desc.m_shapeNum); in DeserializedTensorsFromBuffer()
922 desc.m_shape = new (std::nothrow) int32_t[desc.m_shapeNum]; in DeserializedTensorsFromBuffer()
928 …memRet = memcpy_s(desc.m_shape, desc.m_shapeNum * sizeof(int32_t), ptr, desc.m_shapeNum * sizeof(i… in DeserializedTensorsFromBuffer()
[all …]