Searched refs:CppTensor (Results 1 – 4 of 4) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/system_test/ |
H A D | end_to_end_test.cpp | 47 OH_NN_ReturnCode End2EndTest::BuildModel(const std::vector<CppTensor>& tensors) in BuildModel() 131 CppTensor addend1{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 132 CppTensor addend2{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 135 CppTensor output{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 186 CppTensor addend1{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 187 CppTensor addend2{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 190 CppTensor output{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 257 CppTensor output{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 314 CppTensor output{OH_NN_FLOAT32, {3, 2, 2}, nullptr, 0, quantParam, OH_NN_TENSOR}; 507 CppTensor addend1{OH_NN_INT8, {3, 2, 2}, nullptr, 0, quantParam1, OH_NN_TENSOR}; [all …]
|
H A D | end_to_end_test.h | 33 OH_NN_ReturnCode BuildModel(const std::vector<CppTensor>& tensors);
|
/ohos5.0/foundation/ai/neural_network_runtime/test/system_test/common/ |
H A D | nnrt_test.h | 35 struct CppTensor { struct 53 virtual OH_NN_ReturnCode AddTensors(const std::vector<CppTensor>& cppTensors); argument
|
H A D | nnrt_test.cpp | 49 OH_NN_ReturnCode NNRtTest::AddTensors(const std::vector<CppTensor>& cppTensors) in AddTensors() 53 for (const CppTensor& cppTensor : cppTensors) { in AddTensors()
|