Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionBuilder.h137 bool updateOutputShapes(const std::vector<hal::OutputShape>& outputShapes);
239 bool updateOutputShapes(const std::vector<hal::OutputShape>& from,
DExecutionBuilder.cpp542 if (!executor->updateOutputShapes(stepOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
585 !fallbackExecutor->updateOutputShapes(fallbackOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
899 bool ExecutionBuilder::updateOutputShapes(const std::vector<OutputShape>& outputShapes) { in updateOutputShapes() function in android::nn::ExecutionBuilder
932 if (!updateOutputShapes(outputShapes) || !updateMemories()) { in finishWithoutSyncFence()
956 bool StepExecutor::updateOutputShapes(const std::vector<OutputShape>& from, in updateOutputShapes() function in android::nn::StepExecutor