413                                std::vector<OutputShape>* outputShapes) {  in cpuFallbackPartial()  argument
430     if (!executor->updateOutputShapes(stepOutputShapes, outputShapes)) {  in cpuFallbackPartial()
436             executionCallback->notify(status, *outputShapes, kNoTiming);  in cpuFallbackPartial()
451     std::vector<OutputShape> outputShapes;  in asyncStartComputePartitioned()  local
453     executionBuilder->initializeOutputShapes(&outputShapes);  in asyncStartComputePartitioned()
468             executionCallback->notify(ErrorStatus::NONE, outputShapes, timing);  in asyncStartComputePartitioned()
477                                        &outputShapes)) {  in asyncStartComputePartitioned()
493         if (!executor->updateOutputShapes(stepOutputShapes, &outputShapes)) {  in asyncStartComputePartitioned()
505                                        &outputShapes)) {  in asyncStartComputePartitioned()
514                 executionCallback->notify(status, outputShapes, kNoTiming);  in asyncStartComputePartitioned()
559     auto wrappedFinish = [this](ErrorStatus error, const std::vector<OutputShape>& outputShapes) {  in compute()  argument
560         return finish(error, outputShapes);  in compute()
605 void ExecutionBuilder::initializeOutputShapes(std::vector<OutputShape>* outputShapes) const {  in initializeOutputShapes()
606     outputShapes->resize(mOutputs.size());  in initializeOutputShapes()
608         (*outputShapes)[i].dimensions = mOutputs[i].dimensions;  in initializeOutputShapes()
609         (*outputShapes)[i].isSufficient = true;  in initializeOutputShapes()
624 bool ExecutionBuilder::updateOutputShapes(const std::vector<OutputShape>& outputShapes) {  in updateOutputShapes()  argument
625     if (outputShapes.size() == 0) {  in updateOutputShapes()
628     NN_RET_CHECK_EQ(outputShapes.size(), mOutputs.size());  in updateOutputShapes()
629     for (uint32_t i = 0; i < outputShapes.size(); i++) {  in updateOutputShapes()
631         NN_RET_CHECK(isUpdatable(mOutputs[i].dimensions, outputShapes[i].dimensions));  in updateOutputShapes()
633     for (uint32_t i = 0; i < outputShapes.size(); i++) {  in updateOutputShapes()
634         mOutputs[i].dimensions = outputShapes[i].dimensions;  in updateOutputShapes()
635         mOutputs[i].isSufficient = outputShapes[i].isSufficient;  in updateOutputShapes()
640 ErrorStatus ExecutionBuilder::finish(ErrorStatus, const std::vector<OutputShape>& outputShapes) {  in finish()  argument
643     if (!updateOutputShapes(outputShapes)) {  in finish()
914         auto [status, outputShapes, timing, fallback] =  in startComputeOnDevice()
919             executionCallback->notify(status, outputShapes, timing);  in startComputeOnDevice()
1002     const auto& outputShapes = executor.getOutputShapes();  in computeOnCpu()  local
1003     executionCallback->notify_1_2(convertResultCodeToErrorStatus(err), outputShapes, kNoTiming);  in computeOnCpu()