Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DExecutionCallback.h139 const std::vector<OutputShape>& getOutputShapes() const;
DExecutionCallback.cpp59 const std::vector<OutputShape>& ExecutionCallback::getOutputShapes() const { in getOutputShapes() function in android::nn::ExecutionCallback
DManager.cpp1041 const auto& outputShapes = executor.getOutputShapes(); in computeOnCpu()
/packages/modules/NeuralNetworks/common/include/
DCpuExecutor.h153 const std::vector<OutputShape>& getOutputShapes() const { in getOutputShapes() function
/packages/modules/NeuralNetworks/driver/sample/
DSampleDriver.cpp493 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in asyncExecute()
625 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in executeSynchronouslyBase()
843 convertToV1_2(executor.getOutputShapes()); in execute()
DCanonicalPreparedModel.cpp172 const auto& outputShapes = executor.getOutputShapes(); in execute()
/packages/modules/NeuralNetworks/driver/sample_aidl/
DSampleDriver.cpp450 auto outputShapes = aidl_hal::utils::convert(executor.getOutputShapes()).value(); in executeSynchronously()