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.cpp1088 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_hidl/
DSampleDriver.cpp482 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in asyncExecute()
614 hardware::hidl_vec<V1_2::OutputShape> outputShapes = convertToV1_2(executor.getOutputShapes()); in executeSynchronouslyBase()
832 convertToV1_2(executor.getOutputShapes()); in execute()
/packages/modules/NeuralNetworks/driver/sample/
DCanonicalPreparedModel.cpp167 const auto& outputShapes = executor.getOutputShapes(); in execute()
/packages/modules/NeuralNetworks/driver/sample_aidl/
DSampleDriverAidl.cpp459 auto outputShapes = aidl_hal::utils::convert(executor.getOutputShapes()).value(); in executeSynchronously()