Home
last modified time | relevance | path

Searched refs:Subgraph (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dgraph_analyzer_test.cc46 void ExtendSubgraph(Subgraph* parent) { gran_->ExtendSubgraph(parent); } in ExtendSubgraph()
48 void ExtendSubgraphPortAllOrNone(Subgraph* parent, GenNode* node, in ExtendSubgraphPortAllOrNone()
53 void ExtendSubgraphAllOrNone(Subgraph* parent, GenNode* node) { in ExtendSubgraphAllOrNone()
73 std::deque<Subgraph*>& GetTodo() { return gran_->todo_; } in GetTodo()
154 absl::make_unique<Subgraph>(Subgraph::Identity({GetNode("add2")})); in TEST_F()
175 auto parent = absl::make_unique<Subgraph>(Subgraph::Identity()); in TEST_F()
177 absl::make_unique<Subgraph>(Subgraph::Identity({GetNode("add2")})); in TEST_F()
199 absl::make_unique<Subgraph>(Subgraph::Identity({GetNode("add2")})); in TEST_F()
221 absl::make_unique<Subgraph>(Subgraph::Identity({GetNode("add2")})); in TEST_F()
238 auto root = absl::make_unique<Subgraph>( in TEST_F()
[all …]
Dsubgraph.h32 class Subgraph {
48 explicit Subgraph(Identity id) : id_(std::move(id)), hash_(id_.Hash()) {} in Subgraph() function
51 Subgraph(const Identity& parent_id, GenNode* add_node);
53 Subgraph() = delete;
54 Subgraph(const Subgraph& other) = delete;
55 void operator=(const Subgraph& other) = delete;
58 bool operator<(const Subgraph& other) const { return this->id_ < other.id_; }
60 bool operator==(const Subgraph& other) const {
80 void MergeCollation(const Subgraph& other) { in MergeCollation()
104 explicit SubgraphIterator(const Subgraph::Identity* id);
[all …]
Dsubgraph.cc31 Subgraph::Identity::Identity(InitializerList init) { in Identity()
37 bool Subgraph::Identity::operator<(const Identity& other) const { in operator <()
57 bool Subgraph::Identity::operator==(const Identity& other) const { in operator ==()
70 size_t Subgraph::Identity::Hash() const { in Hash()
79 string Subgraph::Dump() { in Dump()
94 void Subgraph::ExtractForSignature(SigNodeMap* result) { in ExtractForSignature()
112 Subgraph::Subgraph(const Identity& parent_id, GenNode* add_node) in Subgraph() function in tensorflow::grappler::graph_analyzer::Subgraph
120 SubgraphIterator::SubgraphIterator(const Subgraph::Identity* id) in SubgraphIterator()
212 Subgraph* SubgraphPtrSet::ExtendParent(const Subgraph::Identity& parent_id, in ExtendParent()
222 auto sg = absl::make_unique<Subgraph>(parent_id, node); in ExtendParent()
[all …]
Dsubgraph_test.cc50 Subgraph::Identity id1; in TEST()
51 Subgraph::Identity id2; in TEST()
56 Subgraph sg1(id1); in TEST()
57 Subgraph sg2(id2); in TEST()
68 Subgraph::Identity id1; in TEST()
70 Subgraph sg1(id1); in TEST()
94 Subgraph::Identity id; in TEST()
96 Subgraph sg(id); in TEST()
157 Subgraph::Identity id; in TEST()
159 Subgraph sg(id); in TEST()
[all …]
Dgraph_analyzer.h85 void ExtendSubgraph(Subgraph* parent);
90 void ExtendSubgraphAllOrNone(Subgraph* parent, const GenNode* node);
92 void ExtendSubgraphPortAllOrNone(Subgraph* parent, const GenNode* node,
95 void AddExtendedSubgraph(Subgraph* parent, const Subgraph::Identity& id);
99 bool HasInvalidMultiInputs(Subgraph* sg);
121 std::deque<Subgraph*> todo_;
Dgraph_analyzer.cc75 const Subgraph::Identity empty_parent; in FindSubgraphs()
94 void GraphAnalyzer::ExtendSubgraph(Subgraph* parent) { in ExtendSubgraph()
127 Subgraph* sg = sg_set.ExtendParent(parent->id(), neighbor.node); in ExtendSubgraph()
135 void GraphAnalyzer::ExtendSubgraphAllOrNone(Subgraph* parent, in ExtendSubgraphAllOrNone()
137 Subgraph::Identity id = parent->id(); in ExtendSubgraphAllOrNone()
163 void GraphAnalyzer::ExtendSubgraphPortAllOrNone(Subgraph* parent, in ExtendSubgraphPortAllOrNone()
171 Subgraph::Identity id = parent->id(); in ExtendSubgraphPortAllOrNone()
188 void GraphAnalyzer::AddExtendedSubgraph(Subgraph* parent, in AddExtendedSubgraph()
189 const Subgraph::Identity& id) { in AddExtendedSubgraph()
194 auto sg = absl::make_unique<Subgraph>(id); in AddExtendedSubgraph()
[all …]
Dsig_node_test.cc68 Subgraph::Identity id; in BuildSigMap()
72 Subgraph sg(id); in BuildSigMap()
484 Subgraph::Identity id1; in TEST_F()
487 Subgraph sg1(id1); in TEST_F()
502 Subgraph::Identity id2; in TEST_F()
505 Subgraph sg2(id2); in TEST_F()
524 Subgraph::Identity id1; in TEST_F()
527 Subgraph sg1(id1); in TEST_F()
535 Subgraph::Identity id2; in TEST_F()
538 Subgraph sg2(id2); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc112 explicit InterpreterInfo(Subgraph* subgraph) : subgraph_(subgraph) {} in InterpreterInfo()
136 Subgraph* subgraph_;
139 Subgraph::Subgraph(ErrorReporter* error_reporter, in Subgraph() function in tflite::Subgraph
141 std::vector<std::unique_ptr<Subgraph>>* subgraphs) in Subgraph()
166 Subgraph::~Subgraph() { in ~Subgraph()
188 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
191 return static_cast<Subgraph*>(context->impl_) in ReplaceNodeSubsetsWithDelegateKernels()
270 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
325 TfLiteExternalContext* Subgraph::GetExternalContext( in GetExternalContext()
333 TfLiteExternalContext* Subgraph::GetExternalContext( in GetExternalContext()
[all …]
Dsubgraph.h32 class Subgraph {
36 Subgraph(ErrorReporter* error_reporter,
38 std::vector<std::unique_ptr<Subgraph>>* subgraphs);
40 Subgraph(const Subgraph&) = delete;
43 Subgraph(Subgraph&&) = default;
44 Subgraph& operator=(const Subgraph&) = delete;
45 virtual ~Subgraph();
287 std::vector<std::unique_ptr<Subgraph>>* GetSubgraphs() { return subgraphs_; } in GetSubgraphs()
531 std::vector<std::unique_ptr<Subgraph>>* subgraphs_ = nullptr;
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.h39 void BuildAddSubgraph(Subgraph* subgraph);
43 void BuildMulSubgraph(Subgraph* subgraph);
47 void BuildPadSubgraph(Subgraph* subgraph);
54 void BuildIfSubgraph(Subgraph* subgraph);
63 void BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs);
68 void BuildAccumulateLoopBodySubgraph(Subgraph* subgraph);
76 void BuildPadLoopBodySubgraph(Subgraph* subgraph,
81 void BuildWhileSubgraph(Subgraph* subgraph);
84 void CreateConstantInt32Tensor(Subgraph* subgraph, int tensor_index,
Dwhile.cc33 Subgraph* src_subgraph, in CopyTensorsShapeAndType()
35 Subgraph* dst_subgraph, in CopyTensorsShapeAndType()
54 TfLiteStatus CopyTensorsData(TfLiteContext* context, Subgraph* src_subgraph, in CopyTensorsData()
56 Subgraph* dst_subgraph, in CopyTensorsData()
115 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Prepare()
120 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare()
121 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Prepare()
195 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
197 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Eval()
198 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Eval()
Dif.cc63 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Prepare()
68 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
69 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
134 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
142 Subgraph& active_branch_subgraph = in Eval()
Dsubgraph_test_util.cc46 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
60 void SubgraphBuilder::BuildAddSubgraph(Subgraph* subgraph) { in BuildAddSubgraph()
90 void SubgraphBuilder::BuildMulSubgraph(Subgraph* subgraph) { in BuildMulSubgraph()
120 void SubgraphBuilder::BuildPadSubgraph(Subgraph* subgraph) { in BuildPadSubgraph()
148 void SubgraphBuilder::BuildIfSubgraph(Subgraph* subgraph) { in BuildIfSubgraph()
186 void SubgraphBuilder::BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs) { in BuildLessEqualCondSubgraph()
217 void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) { in BuildAccumulateLoopBodySubgraph()
261 void SubgraphBuilder::BuildPadLoopBodySubgraph(Subgraph* subgraph, in BuildPadLoopBodySubgraph()
311 void SubgraphBuilder::BuildWhileSubgraph(Subgraph* subgraph) { in BuildWhileSubgraph()
349 void SubgraphBuilder::CreateConstantInt32Tensor(Subgraph* subgraph, in CreateConstantInt32Tensor()
/external/tensorflow/tensorflow/core/graph/
Dsubgraph_test.cc107 string Subgraph(const string& fed_str, const string& fetch_str, in Subgraph() function in tensorflow::__anon58a5c0460111::SubgraphTest
158 EXPECT_EQ("OK", Subgraph("", "", "t1")); in TEST_F()
171 EXPECT_EQ("OK", Subgraph("", "", "t2,t3_a")); in TEST_F()
184 EXPECT_EQ("OK", Subgraph("input:1", "", "t2")); in TEST_F()
198 Subgraph("input:1", "", "t2", true /* use_function_convention */)); in TEST_F()
207 EXPECT_EQ("OK", Subgraph("W1:0", "", "t1")); in TEST_F()
219 Subgraph("W1:0", "", "t1", true /* use_function_convention */)); in TEST_F()
236 EXPECT_EQ("OK", Subgraph("input:1,t1,W2", "", "t2", in TEST_F()
250 EXPECT_EQ("OK", Subgraph("", "W2,input:1,t1,t2", "t2")); in TEST_F()
264 EXPECT_EQ("OK", Subgraph("", "W2,input:1,t1,t2", "t2", in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc240 class Subgraph { class in tensorflow::__anonbc9ea6eb0211::Encapsulator
653 std::unordered_map<string, Subgraph> subgraphs_;
713 Node* Encapsulator::Subgraph::GetCallNode() const { return call_node_; } in GetCallNode()
715 int Encapsulator::Subgraph::GetArgIndexForEdge(const Edge* edge) const { in GetArgIndexForEdge()
719 int Encapsulator::Subgraph::GetResultIndexForEdge(const Edge* edge) const { in GetResultIndexForEdge()
723 Node* Encapsulator::Subgraph::GetRecvAtHostNode( in GetRecvAtHostNode()
729 int Encapsulator::Subgraph::GetRecvAtHostSlot( in GetRecvAtHostSlot()
735 Node* Encapsulator::Subgraph::GetSendFromHostNode( in GetSendFromHostNode()
741 int Encapsulator::Subgraph::GetSendFromHostSlot( in GetSendFromHostSlot()
747 Node* Encapsulator::Subgraph::MakeNodeImage(const Graph* graph_in, Node* node) { in MakeNodeImage()
[all …]
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate_disabled.cc38 TfLiteStatus NNAPIDelegate::BuildGraph(Subgraph* subgraph) { in BuildGraph()
42 TfLiteStatus NNAPIDelegate::Invoke(Subgraph* subgraph) { return kTfLiteError; } in Invoke()
Dnnapi_delegate.h54 TfLiteStatus BuildGraph(Subgraph* subgraph);
57 TfLiteStatus Invoke(Subgraph* subgraph);
Dinterpreter.h461 Subgraph* subgraph(int subgraph_index) { in subgraph()
469 Subgraph& primary_subgraph() { in primary_subgraph()
474 const Subgraph& primary_subgraph() const { in primary_subgraph()
518 std::vector<std::unique_ptr<Subgraph>> subgraphs_;
Dmodel.h201 Subgraph* subgraph);
205 Subgraph* subgraph);
Dinterpreter.cc113 Subgraph* subgraph = in AddSubgraphs()
114 new Subgraph(error_reporter_, external_contexts_, &subgraphs_); in AddSubgraphs()
Dmodel.cc249 Subgraph* subgraph) { in ParseNodes()
355 Subgraph* subgraph) { in ParseTensors()
527 tflite::Subgraph* modified_subgraph = in operator ()()
Dnnapi_delegate.cc115 TfLiteStatus addTensorOperands(tflite::Subgraph* subgraph, in addTensorOperands()
215 tflite::Subgraph* subgraph, ANeuralNetworksModel* nn_model, in AddOpsAndParams()
698 TfLiteStatus NNAPIDelegate::BuildGraph(Subgraph* subgraph) { in BuildGraph()
771 TfLiteStatus NNAPIDelegate::Invoke(Subgraph* subgraph) { in Invoke()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3.fbs19 // Version 3: Move buffer data from Model.Subgraph.Tensors to Model.Buffers.
Dschema.fbs19 // Version 3: Move buffer data from Model.Subgraph.Tensors to Model.Buffers.