| 301 | } |
| 302 | |
| 303 | tensorflow::Status PersistEagerOutputs(BufferMap* buffer_map) { |
| 304 | auto* handles = outputs_.GetTensorHandles(); |
| 305 | for (int i = 0; i < outputs_.Size(); ++i) { |
| 306 | if (outputs_.IsSubgraphOutput(i)) { |
| 307 | const tensorflow::Tensor* tensor = nullptr; |
| 308 | TF_RETURN_IF_ERROR(handles->at(i)->Tensor(&tensor)); |
| 309 | buffer_map->SetFromTensorFlow(outputs_.TfLiteIndex(i), *tensor); |
| 310 | } |
| 311 | } |
| 312 | return tensorflow::Status::OK(); |
| 313 | } |
| 314 | |
| 315 | private: |
| 316 | OpNode(const OpNode&) = delete; |
no test coverage detected