Searched refs:nnrtDeviceInfo (Results 1 - 2 of 2) sorted by relevance
| /third_party/mindspore/test/unittest/ |
| H A D | nnrt_delegate_unit_test.cpp | 80 auto nnrtDeviceInfo = OH_AI_CreateNNRTDeviceInfoByName(name); in HWTEST() local
81 ASSERT_NE(nnrtDeviceInfo, nullptr); in HWTEST()
83 OH_AI_DeviceType deviceType = OH_AI_DeviceInfoGetDeviceType(nnrtDeviceInfo); in HWTEST()
85 ASSERT_EQ(OH_AI_DeviceInfoGetDeviceId(nnrtDeviceInfo), id); in HWTEST()
87 OH_AI_DeviceInfoDestroy(&nnrtDeviceInfo); in HWTEST()
88 ASSERT_EQ(nnrtDeviceInfo, nullptr); in HWTEST()
91 nnrtDeviceInfo = OH_AI_CreateNNRTDeviceInfoByType(type); in HWTEST()
92 ASSERT_NE(nnrtDeviceInfo, nullptr); in HWTEST()
94 deviceType = OH_AI_DeviceInfoGetDeviceType(nnrtDeviceInfo); in HWTEST()
97 ASSERT_EQ(OH_AI_DeviceInfoGetDeviceId(nnrtDeviceInfo), i in HWTEST()
[all...] |
| /third_party/mindspore/test/utils/ |
| H A D | model_utils.cpp | 82 OH_AI_DeviceInfoHandle nnrtDeviceInfo = OH_AI_DeviceInfoCreate(OH_AI_DEVICETYPE_NNRT); in AddContextDeviceNNRT() local
83 ASSERT_NE(nnrtDeviceInfo, nullptr); in AddContextDeviceNNRT()
84 OH_AI_DeviceInfoSetDeviceId(nnrtDeviceInfo, id); in AddContextDeviceNNRT()
87 OH_AI_DeviceType deviceType = OH_AI_DeviceInfoGetDeviceType(nnrtDeviceInfo); in AddContextDeviceNNRT()
91 OH_AI_DeviceInfoSetPerformanceMode(nnrtDeviceInfo, OH_AI_PERFORMANCE_MEDIUM); in AddContextDeviceNNRT()
92 ASSERT_EQ(OH_AI_DeviceInfoGetPerformanceMode(nnrtDeviceInfo), OH_AI_PERFORMANCE_MEDIUM); in AddContextDeviceNNRT()
93 OH_AI_DeviceInfoSetPriority(nnrtDeviceInfo, OH_AI_PRIORITY_MEDIUM); in AddContextDeviceNNRT()
94 ASSERT_EQ(OH_AI_DeviceInfoGetPriority(nnrtDeviceInfo), OH_AI_PRIORITY_MEDIUM); in AddContextDeviceNNRT()
96 OH_AI_ContextAddDeviceInfo(context, nnrtDeviceInfo); in AddContextDeviceNNRT()
|
Completed in 1 milliseconds