Home
last modified time | relevance | path

Searched refs:mPreparedModel (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/runtime/
DCallbacks.cpp102 mErrorStatus(ErrorStatus::GENERAL_FAILURE), mPreparedModel(nullptr) {} in PreparedModelCallback()
109 mPreparedModel = preparedModel; in notify()
121 return mPreparedModel; in getPreparedModel()
DExecutionBuilder.cpp462 mDriver(driver), mPreparedModel(preparedModel), in StepExecutor()
550 if (mPreparedModel == nullptr) { in startComputeOnDevice()
572 mPreparedModel = preparedModelCallback->getPreparedModel(); in startComputeOnDevice()
576 if (mPreparedModel == nullptr) { in startComputeOnDevice()
641 Return<ErrorStatus> executeStatus = mPreparedModel->execute(request, executionCallback); in startComputeOnDevice()
DCallbacks.h248 sp<IPreparedModel> mPreparedModel; variable
DExecutionBuilder.h186 sp<IPreparedModel> mPreparedModel; // nullptr if CPU execution or if bypassing ExecutionPlan variable
DExecutionPlan.h246 sp<IPreparedModel> mPreparedModel; // not used for CPU member
DExecutionPlan.cpp474 const int n = compile(mDevice, mModel, executionPreference, &mPreparedModel); in finish()
608 simpleBody->mPreparedModel); in next()
/frameworks/ml/nn/runtime/test/
DTestExecution.cpp60 mPreparedModel(preparedModel), mErrorStatus(errorStatus) {} in TestPreparedModel()
65 return mPreparedModel->execute(request, callback); in execute()
72 sp<IPreparedModel> mPreparedModel; member in android::__anon50a6e8490111::TestPreparedModel