Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc682 if (memory_planner_ && !memory_planner_->HasNonPersistentMemory()) { in AllocateTensors()
687 memory_planner_->AcquireNonPersistentMemory(); in AllocateTensors()
695 if (memory_planner_) { in AllocateTensors()
696 TF_LITE_ENSURE_STATUS(memory_planner_->ResetAllocations()); in AllocateTensors()
873 if (memory_planner_) { in ReleaseNonPersistentMemory()
874 TF_LITE_ENSURE_STATUS(memory_planner_->ReleaseNonPersistentMemory()); in ReleaseNonPersistentMemory()
935 if (!memory_planner_) { in PrepareOpsAndTensors()
936 memory_planner_.reset(new ArenaPlanner( in PrepareOpsAndTensors()
940 memory_planner_->PlanAllocations(); in PrepareOpsAndTensors()
973 TF_LITE_ENSURE_STATUS(memory_planner_->ExecuteAllocations( in PrepareOpsAndTensors()
[all …]
Dsubgraph.h710 std::unique_ptr<MemoryPlanner> memory_planner_; variable