Home
last modified time | relevance | path

Searched refs:execution_plan_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc398 execution_plan_.clear(); in ReplaceNodeSubsetsWithDelegateKernels()
408 execution_plan_.push_back(*it); in ReplaceNodeSubsetsWithDelegateKernels()
471 plan_cache_.reset(TfLiteIntArrayCreate(execution_plan_.size())); in GetExecutionPlan()
473 static_assert(sizeof(plan_cache_->data[0]) == sizeof(execution_plan_[0]), in GetExecutionPlan()
475 std::memcpy(plan_cache_->data, execution_plan_.data(), in GetExecutionPlan()
476 sizeof(plan_cache_->data[0]) * execution_plan_.size()); in GetExecutionPlan()
804 execution_plan_.push_back(new_node_index); in AddNodeWithParameters()
969 execution_plan_, &last_exec_plan_index_prepared)); in PrepareOpsAndTensors()
1017 execution_plan_index < execution_plan_.size(); execution_plan_index++) { in Invoke()
1023 int node_index = execution_plan_[execution_plan_index]; in Invoke()
[all …]
Dsubgraph.h192 std::vector<int>& execution_plan() { return execution_plan_; } in execution_plan()
195 const std::vector<int>& execution_plan() const { return execution_plan_; } in execution_plan()
690 std::vector<int> execution_plan_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel.cc120 execution_plan_.push_back(new_id); in NewNode()
129 while (idx < execution_plan_.size()) { in InsertNodeAfter()
130 if (execution_plan_[idx] == id) break; in InsertNodeAfter()
133 if (idx == execution_plan_.size()) { in InsertNodeAfter()
142 execution_plan_.insert(execution_plan_.begin() + idx + 1, new_id); in InsertNodeAfter()
309 model->execution_plan_.clear(); in MakeExactCopy()
318 for (auto node_id : execution_plan_) { in MakeExactCopy()
319 model->execution_plan_.push_back(node_id); in MakeExactCopy()
327 for (auto node_id : execution_plan_) { in MakeExactCopy()
Dmodel.h209 for (const auto id : execution_plan_) { in FilterNodes()
225 std::vector<NodeId> execution_plan_; variable
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.h92 execution_plan_(subgraph->execution_plan()) { in SubgraphWriter()
127 execution_plan_(subgraph->execution_plan()) { in SubgraphWriter()
191 std::vector<int> execution_plan_; variable
Dwriter_lib.cc94 for (int op_index : execution_plan_) { in ExportOperators()
107 for (int op_index : execution_plan_) { in ExportOperators()
392 execution_plan_ = execution_plan; in SetCustomInputOutput()