MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / PersistEagerOutputs

Method PersistEagerOutputs

tensorflow/lite/delegates/flex/kernel.cc:303–313  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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;

Callers 1

ExecuteFlexOpFunction · 0.80

Calls 7

GetTensorHandlesMethod · 0.80
IsSubgraphOutputMethod · 0.80
SetFromTensorFlowMethod · 0.80
SizeMethod · 0.45
TensorMethod · 0.45
atMethod · 0.45
TfLiteIndexMethod · 0.45

Tested by

no test coverage detected