Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
Dinterpreter.cc190 const size_t base_index = subgraphs_.size(); in AddSubgraphs()
193 subgraphs_.reserve(base_index + subgraphs_to_add); in AddSubgraphs()
196 error_reporter_, external_contexts_, &subgraphs_, &resources_, in AddSubgraphs()
197 &resource_ids_, &initialization_status_map_, subgraphs_.size()); in AddSubgraphs()
198 subgraphs_.emplace_back(subgraph); in AddSubgraphs()
301 for (auto& subgraph : subgraphs_) { in SetNumThreads()
390 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegateImpl()
408 for (auto& subgraph : subgraphs_) { in RemoveAllDelegates()
417 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetMetadata()
419 TF_LITE_ENSURE_STATUS(subgraphs_[subgraph_index]->SetMetadata(&metadata_)); in SetMetadata()
[all …]
Dinterpreter_experimental.cc52 for (auto& subgraph : subgraphs_) { in ResetVariableTensors()
59 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32()
68 for (auto& subgraph : subgraphs_) { in SetCancellationFunction()
Dinterpreter.h701 size_t subgraphs_size() const { return subgraphs_.size(); } in subgraphs_size()
710 return subgraphs_[subgraph_index].get(); in subgraph()
721 return *subgraphs_.front(); /// Safe as subgraphs_ always has 1 entry. in primary_subgraph()
726 return *subgraphs_.front(); // Safe as subgraphs_ always has 1 entry. in primary_subgraph()
871 std::vector<std::unique_ptr<Subgraph>> subgraphs_; variable
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Dpick_subgraphs.cc174 llvm::DenseMap<Operation*, Subgraph> subgraphs_; member in mlir::TFL::tac::__anon83dfcf2f0111::PickSubgraphsPass
223 Subgraph& current_subgraph = subgraphs_.find(current_graph_call)->second; in ProcessSubgraph()
236 Subgraph& input_subgraph = subgraphs_.find(input_call)->second; in ProcessSubgraph()
339 subgraphs_.try_emplace(call_op, subgraph); in BuildSubgraphs()
377 const Subgraph& subgraph = subgraphs_.find(output)->second; in MakeDecisions()
401 const Subgraph& input_subgraph = subgraphs_.find(input)->second; in MakeDecisions()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc360 std::unordered_map<string, Subgraph> subgraphs_; member in tensorflow::__anon018d25a70111::Encapsulator
672 Subgraph& subgraph = subgraphs_[func_id]; in CopySubgraphNodes()
694 Graph* g = subgraphs_[src_func_id].GetGraph(); in CopySubgraphEdges()
716 Subgraph& src_subgraph = subgraphs_[src_func_id]; in CopySubgraphEdges()
739 Subgraph& dst_subgraph = subgraphs_[dst_func_id]; in CopySubgraphEdges()
767 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
774 for (auto& entry : subgraphs_) { in SplitIntoSubgraphs()
787 for (auto& subgraph_entry : subgraphs_) { in BuildFunctionDefs()
816 for (auto& subgraph_entry : subgraphs_) { in AddFunctionCallNodes()
830 *src_image = subgraphs_.at(src_func_id).GetCallNode(); in FindOutputImageOfEdgeSrc()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h333 std::vector<std::unique_ptr<Subgraph>>* GetSubgraphs() { return subgraphs_; } in GetSubgraphs()
857 std::vector<std::unique_ptr<Subgraph>>* subgraphs_ = nullptr; variable
Dsubgraph.cc242 subgraphs_(subgraphs), in Subgraph()