Searched refs:m_format (Results 1 – 8 of 8) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | nn_tensor.cpp | 57 m_format = tensor.m_format; in operator =()
147 m_format = nnTensorInfo.format; in BuildFromOHNNTensorInfo()
351 m_format = format; in SetFormat()
416 return m_format; in GetFormat()
427 mindspore::lite::Format format = NNToMS::TransformFormat(m_format); in ConvertToLiteGraphTensor()
453 tensor.format = m_format; in ConvertToIOTensor()
462 desc.SetFormat(m_format); in ConvertToTensorDesc()
494 if (m_format != tensor.GetFormat()) { in CompareAttribute()
495 LOGI("Tensors have different format: %d and %d.", m_format, tensor.GetFormat()); in CompareAttribute()
|
| H A D | nncompiler.cpp | 56 ret = tensorDesc.first->GetFormat(&m_format); in CopyFromTensorDesc()
87 ret = tensorDesc.SetFormat(m_format); in CopyToTensorDesc()
109 OH_NN_Format m_format{OH_NN_FORMAT_NONE}; member
117 const size_t SIZE_OF_FORMAT = sizeof(SerializedTensorDesc::m_format);
825 memRet = memcpy_s(currentPos, SIZE_OF_FORMAT, &tensorDesc.m_format, SIZE_OF_FORMAT); in SerializeTensorsToBuffer()
898 memRet = memcpy_s(&desc.m_format, SIZE_OF_FORMAT, ptr, sizeof(desc.m_format)); in DeserializedTensorsFromBuffer()
904 ptr += sizeof(desc.m_format); in DeserializedTensorsFromBuffer()
|
| H A D | nn_tensor.h | 91 OH_NN_Format m_format {OH_NN_FORMAT_NHWC};
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/ |
| H A D | tensor_desc.cpp | 78 *format = m_format; in GetFormat()
88 m_format = format; in SetFormat()
|
| H A D | tensor_desc.h | 47 OH_NN_Format m_format {OH_NN_FORMAT_NONE};
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | maxpool_builder.cpp | 50 m_padMode, m_format, m_global, m_activationType); in GetPrimitive()
|
| H A D | avgpool_builder.cpp | 51 m_padMode, m_roundMode, m_format, m_global, m_activationType); in GetPrimitive()
|
| H A D | pooling_builder.h | 56 mindspore::lite::Format m_format {mindspore::lite::FORMAT_NCHW};
|