Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionBuilder.cpp389 sp<ExecutionCallback> fallbackCallback; in cpuFallbackFull() local
390 int n = executor.startCompute(&fallbackCallback); in cpuFallbackFull()
395 fallbackCallback->wait(); in cpuFallbackFull()
396 executionCallback->notify(fallbackCallback->getStatus(), fallbackCallback->getOutputShapes(), in cpuFallbackFull()
397 fallbackCallback->getTiming()); in cpuFallbackFull()
422 sp<ExecutionCallback> fallbackCallback; in cpuFallbackPartial() local
423 if (executor->startComputeOnCpu(&fallbackCallback) != ANEURALNETWORKS_NO_ERROR) { in cpuFallbackPartial()
427 fallbackCallback->wait(); in cpuFallbackPartial()
428 ErrorStatus status = fallbackCallback->getStatus(); in cpuFallbackPartial()
429 const auto& stepOutputShapes = fallbackCallback->getOutputShapes(); in cpuFallbackPartial()