Home
last modified time | relevance | path

Searched refs:OH_AI_KernelCallBack (Results 1 - 2 of 2) sorted by relevance

/third_party/mindspore/interfaces/kits/c/
H A Dmodel.h66 typedef bool (*OH_AI_KernelCallBack)(const OH_AI_TensorHandleArray inputs, const OH_AI_TensorHandleArray outputs, typedef
131 OH_AI_TensorHandleArray *outputs, const OH_AI_KernelCallBack before,
132 const OH_AI_KernelCallBack after);
254 OH_AI_API OH_AI_Status OH_AI_RunStep(OH_AI_ModelHandle model, const OH_AI_KernelCallBack before,
255 const OH_AI_KernelCallBack after);
/third_party/mindspore/test/utils/
H A Dmodel_utils.cpp195 OH_AI_KernelCallBack beforeCallBack = PrintBeforeCallback; in ModelPredict()
196 OH_AI_KernelCallBack afterCallBack = PrintAfterCallback; in ModelPredict()
273 OH_AI_KernelCallBack beforeCallBack = PrintBeforeCallback; in ModelTrain()
274 OH_AI_KernelCallBack afterCallBack = PrintAfterCallback; in ModelTrain()

Completed in 2 milliseconds