Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DCallbacks.h295 const std::vector<OutputShape>& getOutputShapes() const;
DCallbacks.cpp125 const std::vector<OutputShape>& ExecutionCallback::getOutputShapes() const { in getOutputShapes() function in android::hardware::neuralnetworks::V1_2::implementation::ExecutionCallback
DExecutionBuilder.cpp396 executionCallback->notify(fallbackCallback->getStatus(), fallbackCallback->getOutputShapes(), in cpuFallbackFull()
429 const auto& stepOutputShapes = fallbackCallback->getOutputShapes(); in cpuFallbackPartial()
492 const auto& stepOutputShapes = stepCallback->getOutputShapes(); in asyncStartComputePartitioned()
1002 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu()
DVersionedInterfaces.cpp206 return {callback->getStatus(), callback->getOutputShapes(), callback->getTiming()}; in executeSynchronously()
/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp248 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute()
335 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronously()
410 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in execute()
/frameworks/ml/nn/common/include/
DCpuExecutor.h147 const std::vector<OutputShape>& getOutputShapes() const { in getOutputShapes() function