Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnn_tensor.cpp56 m_dataType = tensor.m_dataType; in operator =()
86 m_dataType = dataType; in Build()
113 m_dataType = nnTensor.dataType; in BuildFromOHNNTensor()
141 m_dataType = nnTensorInfo.dataType; in BuildFromOHNNTensorInfo()
235 uint64_t dataLength {static_cast<uint64_t>(GetTypeSize(m_dataType))}; in ValidateDimensions()
401 return m_dataType; in GetDataType()
426 mindspore::lite::DataType dataType = NNToMS::TransformDataType(m_dataType); in ConvertToLiteGraphTensor()
452 tensor.dataType = m_dataType; in ConvertToIOTensor()
461 desc.SetDataType(m_dataType); in ConvertToTensorDesc()
489 if (m_dataType != tensor.GetDataType()) { in CompareAttribute()
[all …]
H A Dnncompiler.cpp50 OH_NN_ReturnCode ret = tensorDesc.first->GetDataType(&m_dataType); in CopyFromTensorDesc()
81 OH_NN_ReturnCode ret = tensorDesc.SetDataType(m_dataType); in CopyToTensorDesc()
108 OH_NN_DataType m_dataType{OH_NN_UNKNOWN}; member
116 const size_t SIZE_OF_DATATYPE = sizeof(SerializedTensorDesc::m_dataType);
817 … auto memRet = memcpy_s(currentPos, SIZE_OF_DATATYPE, &tensorDesc.m_dataType, SIZE_OF_DATATYPE); in SerializeTensorsToBuffer()
890 auto memRet = memcpy_s(&desc.m_dataType, SIZE_OF_DATATYPE, ptr, sizeof(desc.m_dataType)); in DeserializedTensorsFromBuffer()
896 ptr += sizeof(desc.m_dataType); in DeserializedTensorsFromBuffer()
H A Dnn_tensor.h90 OH_NN_DataType m_dataType {OH_NN_FLOAT32};
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dtensor_desc.cpp58 *dataType = m_dataType; in GetDataType()
68 m_dataType = dataType; in SetDataType()
167 uint32_t typeSize = GetTypeSize(m_dataType); in GetByteSize()
H A Dtensor_desc.h46 OH_NN_DataType m_dataType {OH_NN_UNKNOWN};
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dconstant_of_shape_builder.cpp48 m_dataType = *static_cast<const int64_t*>(buffer); in SetDataType()
131 void* primitive = mindspore::lite::MindIR_ConstantOfShape_CreatePrimitive(m_dataType, m_value); in GetPrimitive()
H A Dconstant_of_shape_builder.h45 int64_t m_dataType {0};