Searched refs:m_shape (Results 1 – 5 of 5) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/ |
| H A D | tensor_desc.cpp | 106 *shape = const_cast<int32_t*>(m_shape.data()); in GetShape()
107 *shapeNum = m_shape.size(); in GetShape()
121 m_shape.clear(); in SetShape()
123 m_shape.emplace_back(shape[i]); in SetShape()
134 if (m_shape.empty()) { in GetElementNum()
139 size_t shapeNum = m_shape.size(); in GetElementNum()
141 if (m_shape[i] <= 0) { in GetElementNum()
142 …OGW("GetElementNum return 0 with dynamic shape, shape[%{public}zu] is %{public}d.", i, m_shape[i]); in GetElementNum()
146 (*elementNum) *= m_shape[i]; in GetElementNum()
|
| H A D | tensor_desc.h | 48 std::vector<int32_t> m_shape; variable
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | broadcast_to_builder.cpp | 37 m_shape.clear(); in SetShape()
49 m_shape.emplace_back(*pShape); in SetShape()
108 void* primitive = mindspore::lite::MindIR_BroadcastTo_CreatePrimitive(m_shape); in GetPrimitive()
|
| H A D | broadcast_to_builder.h | 44 std::vector<int64_t> m_shape;
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | nncompiler.cpp | 62 ret = tensorDesc.first->GetShape(&m_shape, &m_shapeNum); in CopyFromTensorDesc()
93 ret = tensorDesc.SetShape(m_shape, m_shapeNum); in CopyToTensorDesc()
112 int32_t* m_shape{nullptr}; member
850 memRet = memcpy_s(currentPos, sizeOfShape, tensorDesc.m_shape, sizeOfShape); in SerializeTensorsToBuffer()
876 delete[] desc.m_shape; in ReleaseDescShape()
922 desc.m_shape = new (std::nothrow) int32_t[desc.m_shapeNum]; in DeserializedTensorsFromBuffer()
923 if (desc.m_shape == nullptr) { in DeserializedTensorsFromBuffer()
928 …memRet = memcpy_s(desc.m_shape, desc.m_shapeNum * sizeof(int32_t), ptr, desc.m_shapeNum * sizeof(i… in DeserializedTensorsFromBuffer()
|