Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionBuilder.h127 bool updateOutputShapes(const std::vector<OutputShape>& outputShapes);
189 bool updateOutputShapes(const std::vector<OutputShape>& from, std::vector<OutputShape>* to);
DExecutionBuilder.cpp430 if (!executor->updateOutputShapes(stepOutputShapes, outputShapes)) { in cpuFallbackPartial()
493 if (!executor->updateOutputShapes(stepOutputShapes, &outputShapes)) { in asyncStartComputePartitioned()
624 bool ExecutionBuilder::updateOutputShapes(const std::vector<OutputShape>& outputShapes) { in updateOutputShapes() function in android::nn::ExecutionBuilder
643 if (!updateOutputShapes(outputShapes)) { in finish()
649 bool StepExecutor::updateOutputShapes(const std::vector<OutputShape>& from, in updateOutputShapes() function in android::nn::StepExecutor