Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DCallbacks.h388 const std::vector<hal::OutputShape>& getOutputShapes() const;
DCallbacks.cpp142 const std::vector<OutputShape>& ExecutionCallback::getOutputShapes() const { in getOutputShapes() function in android::nn::ExecutionCallback
DManager.cpp707 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu()
DVersionedInterfaces.cpp218 return getExecutionResult(cb.getStatus(), cb.getOutputShapes(), cb.getTiming()); in executeAsynchronously()
/frameworks/ml/nn/common/include/
DCpuExecutor.h153 const std::vector<hal::OutputShape>& getOutputShapes() const { in getOutputShapes() function
/frameworks/ml/nn/driver/sample/
DSampleDriver.cpp482 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in asyncExecute()
607 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in executeSynchronouslyBase()
819 hidl_vec<OutputShape> outputShapes = executor.getOutputShapes(); in execute()