Home
last modified time | relevance | path

Searched refs:subgraphs_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/
Dinterpreter.cc108 const size_t base_index = subgraphs_.size(); in AddSubgraphs()
111 subgraphs_.reserve(base_index + subgraphs_to_add); in AddSubgraphs()
114 new Subgraph(error_reporter_, external_contexts_, &subgraphs_); in AddSubgraphs()
115 subgraphs_.emplace_back(subgraph); in AddSubgraphs()
207 for (auto& subgraph : subgraphs_) { in SetNumThreads()
220 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32()
229 for (auto& subgraph : subgraphs_) { in SetCancellationFunction()
235 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegate()
275 for (auto& subgraph : subgraphs_) subgraph->SetProfiler(profiler); in SetProfiler()
Dinterpreter.h457 size_t subgraphs_size() const { return subgraphs_.size(); } in subgraphs_size()
465 return &*subgraphs_[subgraph_index]; in subgraph()
470 return *subgraphs_.front(); // Safe as subgraphs_ always has 1 entry. in primary_subgraph()
475 return *subgraphs_.front(); // Safe as subgraphs_ always has 1 entry. in primary_subgraph()
518 std::vector<std::unique_ptr<Subgraph>> subgraphs_; variable
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc653 std::unordered_map<string, Subgraph> subgraphs_; member in tensorflow::__anonbc9ea6eb0211::Encapsulator
1425 Subgraph& subgraph = subgraphs_[func_id]; in CopySubgraphNodes()
1452 Graph* g = subgraphs_[src_func_id].GetGraph(); in CopySubgraphEdges()
1474 Subgraph& src_subgraph = subgraphs_[src_func_id]; in CopySubgraphEdges()
1504 Subgraph& dst_subgraph = subgraphs_[dst_func_id]; in CopySubgraphEdges()
1543 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
1550 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
1557 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
1570 for (auto& subgraph_entry : subgraphs_) { in BuildFunctionDefs()
1601 for (auto& subgraph_entry : subgraphs_) { in AddFunctionCallNodes()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h287 std::vector<std::unique_ptr<Subgraph>>* GetSubgraphs() { return subgraphs_; } in GetSubgraphs()
531 std::vector<std::unique_ptr<Subgraph>>* subgraphs_ = nullptr; variable
Dsubgraph.cc146 subgraphs_(subgraphs) { in Subgraph()