Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
Dinterpreter.cc244 const size_t base_index = subgraphs_.size(); in AddSubgraphs()
247 subgraphs_.reserve(base_index + subgraphs_to_add); in AddSubgraphs()
250 &subgraphs_, &resources_); in AddSubgraphs()
251 subgraphs_.emplace_back(subgraph); in AddSubgraphs()
357 for (auto& subgraph : subgraphs_) { in SetNumThreads()
371 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32()
380 for (auto& subgraph : subgraphs_) { in SetCancellationFunction()
389 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegate()
407 for (auto& subgraph : subgraphs_) { in RemoveAllDelegates()
461 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetSubgraphProfiler()
[all …]
Dinterpreter.h635 size_t subgraphs_size() const { return subgraphs_.size(); } in subgraphs_size()
643 return &*subgraphs_[subgraph_index]; in subgraph()
648 return *subgraphs_.front(); /// Safe as subgraphs_ always has 1 entry. in primary_subgraph()
653 return *subgraphs_.front(); // Safe as subgraphs_ always has 1 entry. in primary_subgraph()
759 std::vector<std::unique_ptr<Subgraph>> subgraphs_; variable
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc379 std::unordered_map<string, Subgraph> subgraphs_; member in tensorflow::__anondf15e6490211::Encapsulator
698 Subgraph& subgraph = subgraphs_[func_id]; in CopySubgraphNodes()
720 Graph* g = subgraphs_[src_func_id].GetGraph(); in CopySubgraphEdges()
742 Subgraph& src_subgraph = subgraphs_[src_func_id]; in CopySubgraphEdges()
765 Subgraph& dst_subgraph = subgraphs_[dst_func_id]; in CopySubgraphEdges()
793 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
800 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
813 for (auto& subgraph_entry : subgraphs_) { in BuildFunctionDefs()
842 for (auto& subgraph_entry : subgraphs_) { in AddFunctionCallNodes()
856 *src_image = subgraphs_.at(src_func_id).GetCallNode(); in FindOutputImageOfEdgeSrc()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h320 std::vector<std::unique_ptr<Subgraph>>* GetSubgraphs() { return subgraphs_; } in GetSubgraphs()
724 std::vector<std::unique_ptr<Subgraph>>* subgraphs_ = nullptr; variable
Dsubgraph.cc217 subgraphs_(subgraphs), in Subgraph()