Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dtensor_desc.cpp92 OH_NN_ReturnCode TensorDesc::GetShape(int32_t** shape, size_t* shapeNum) const in GetShape()
102 if (shapeNum == nullptr) { in GetShape()
107 *shapeNum = m_shape.size(); in GetShape()
111 OH_NN_ReturnCode TensorDesc::SetShape(const int32_t* shape, size_t shapeNum) in SetShape() argument
117 if (shapeNum == 0) { in SetShape()
122 for (size_t i = 0; i < shapeNum; ++i) { in SetShape()
139 size_t shapeNum = m_shape.size(); in GetElementNum() local
140 for (size_t i = 0; i < shapeNum; ++i) { in GetElementNum()
H A Dtensor_desc.h36 OH_NN_ReturnCode GetShape(int32_t** shape, size_t* shapeNum) const;
37 OH_NN_ReturnCode SetShape(const int32_t* shape, size_t shapeNum);
H A Dexecutor.h39 size_t* shapeNum) const = 0;
40 …_NN_ReturnCode GetOutputShape(uint32_t outputIndex, int32_t** shape, uint32_t* shapeNum) const = 0;
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_tensor/
H A Dnn_tensor_test.cpp292 size_t shapeNum = 1; variable
395 size_t shapeNum = 1; variable
460 size_t shapeNum = 1; variable
529 size_t shapeNum = 1; variable
596 size_t shapeNum = 1; variable
625 size_t shapeNum = 1; variable
776 size_t shapeNum = 1; variable
807 size_t shapeNum = 1; variable
838 size_t shapeNum = 1; variable
869 size_t shapeNum = 1; variable
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnnexecutor.cpp72 size_t inputIndex, size_t** minInputDims, size_t** maxInputDims, size_t* shapeNum) const in GetInputDimRange()
82 if (shapeNum == nullptr) { in GetInputDimRange()
100 *shapeNum = m_minInputDimsVec[inputIndex].size(); in GetInputDimRange()
101 if (*shapeNum != m_maxInputDimsVec[inputIndex].size()) { in GetInputDimRange()
111 …urnCode NNExecutor::GetOutputShape(uint32_t outputIndex, int32_t** shape, uint32_t* shapeNum) const in GetOutputShape()
129 *shapeNum = static_cast<uint32_t>(shapeNumTmp); in GetOutputShape()
398 size_t shapeNum {0}; in CompareAttribute() local
399 ret = tensorDesc.first->GetShape(&shape, &shapeNum); in CompareAttribute()
405 if (shapeNum != dimensions.size()) { in CompareAttribute()
406 LOGI("Tensors have differents dimension counts: %zu and %zu.", shapeNum, dimensions.size()); in CompareAttribute()
H A Dhdi_prepared_model_v1_0.cpp139 size_t shapeNum = 0; in TransIOTensor() local
140 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
146 for (size_t i = 0; i < shapeNum; ++i) { in TransIOTensor()
H A Dhdi_prepared_model_v2_0.cpp140 size_t shapeNum = 0; in TransIOTensor() local
141 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
147 for (size_t i = 0; i < shapeNum; ++i) { in TransIOTensor()
H A Dhdi_prepared_model_v2_1.cpp140 size_t shapeNum = 0; in TransIOTensor() local
141 ret = nnTensorDesc->GetShape(&shape, &shapeNum); in TransIOTensor()
147 for (size_t i = 0; i < shapeNum; ++i) { in TransIOTensor()
H A Dnnexecutor.h38 size_t* shapeNum) const override;
39 …OH_NN_ReturnCode GetOutputShape(uint32_t outputIndex, int32_t** shape, uint32_t* shapeNum) const o…
H A Dnn_tensor.cpp182 size_t shapeNum {0}; in BuildFromTensorDesc() local
183 returnCode = tensorDescImpl->GetShape(&shape, &shapeNum); in BuildFromTensorDesc()
188 std::vector<int32_t> dimensions(shape, shape + shapeNum); in BuildFromTensorDesc()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/nn_executor/
H A Dnn_executor_test.cpp473 uint32_t* shapeNum = &dimensionCount; variable
474 OH_NN_ReturnCode ret = nnExecutor->GetOutputShape(m_index, dimensions, shapeNum);
502 uint32_t* shapeNum = &dimensionCount; variable
503 OH_NN_ReturnCode ret = nnExecutor->GetOutputShape(m_index, dimensions, shapeNum);
533 uint32_t* shapeNum = &dimensionCount; variable
534 OH_NN_ReturnCode ret = nnExecutor->GetOutputShape(m_index, dimensions, shapeNum);
569 uint32_t* shapeNum = &dimensionCount; variable
571 OH_NN_ReturnCode ret = nnExecutor->GetOutputShape(m_index, dimensions, shapeNum);
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp243 uint32_t* shapeNum = &outputIndex; in SetInputAndOutput() local
244 EXPECT_EQ(OH_NN_SUCCESS, executor.GetOutputShape(outputIndex, shapeAA, shapeNum)); in SetInputAndOutput()
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp245 uint32_t* shapeNum = &outputIndex; in SetInputAndOutput() local
246 EXPECT_EQ(OH_NN_SUCCESS, executor.GetOutputShape(outputIndex, shapeAA, shapeNum)); in SetInputAndOutput()