Home
last modified time | relevance | path

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

12

/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.h33 class Subgraph {
49 explicit Subgraph(Identity id) : id_(std::move(id)), hash_(id_.Hash()) {} in Subgraph() function
52 Subgraph(const Identity& parent_id, GenNode* add_node);
54 Subgraph() = delete;
55 Subgraph(const Subgraph& other) = delete;
56 void operator=(const Subgraph& other) = delete;
59 bool operator<(const Subgraph& other) const { return this->id_ < other.id_; }
61 bool operator==(const Subgraph& other) const {
81 void MergeCollation(const Subgraph& other) { in MergeCollation()
105 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()
214 Subgraph* SubgraphPtrSet::ExtendParent(const Subgraph::Identity& parent_id, in ExtendParent()
224 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()
128 Subgraph* sg = sg_set.ExtendParent(parent->id(), neighbor.node); in ExtendSubgraph()
136 void GraphAnalyzer::ExtendSubgraphAllOrNone(Subgraph* parent, in ExtendSubgraphAllOrNone()
138 Subgraph::Identity id = parent->id(); in ExtendSubgraphAllOrNone()
165 void GraphAnalyzer::ExtendSubgraphPortAllOrNone(Subgraph* parent, in ExtendSubgraphPortAllOrNone()
173 Subgraph::Identity id = parent->id(); in ExtendSubgraphPortAllOrNone()
191 void GraphAnalyzer::AddExtendedSubgraph(Subgraph* parent, in AddExtendedSubgraph()
192 const Subgraph::Identity& id) { in AddExtendedSubgraph()
197 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/kernels/
Dsubgraph_test_util.h44 void BuildAddSubgraph(Subgraph* subgraph);
48 void BuildMulSubgraph(Subgraph* subgraph);
52 void BuildPadSubgraph(Subgraph* subgraph);
59 void BuildIfSubgraph(Subgraph* subgraph);
68 void BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs);
73 void BuildAccumulateLoopBodySubgraph(Subgraph* subgraph);
81 void BuildPadLoopBodySubgraph(Subgraph* subgraph,
86 void BuildWhileSubgraph(Subgraph* subgraph);
90 void BuildAssignRandomValueToVariableSubgraph(Subgraph* graph);
94 void BuildCallOnceAndReadVariableSubgraph(Subgraph* graph);
[all …]
Dcall_once.cc68 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Prepare()
73 Subgraph* init_subgraph = (*subgraphs)[op_data->init_subgraph_index].get(); in Prepare()
85 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
87 Subgraph& init_subgraph = *(*subgraphs)[op_data->init_subgraph_index]; in Eval()
Dwhile.cc45 Subgraph* src_subgraph, in CopyTensorsShapeAndType()
47 Subgraph* dst_subgraph, in CopyTensorsShapeAndType()
74 TfLiteStatus CopyTensorsData(TfLiteContext* context, Subgraph* src_subgraph, in CopyTensorsData()
76 Subgraph* dst_subgraph, in CopyTensorsData()
137 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Prepare()
142 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare()
143 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Prepare()
216 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
218 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Eval()
219 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Eval()
Dif.cc70 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Prepare()
75 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
76 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
144 Subgraph* this_subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
152 Subgraph& active_branch_subgraph = in Eval()
Dsubgraph_test_util.cc81 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
95 void SubgraphBuilder::BuildAddSubgraph(Subgraph* subgraph) { in BuildAddSubgraph()
126 void SubgraphBuilder::BuildMulSubgraph(Subgraph* subgraph) { in BuildMulSubgraph()
157 void SubgraphBuilder::BuildPadSubgraph(Subgraph* subgraph) { in BuildPadSubgraph()
186 void SubgraphBuilder::BuildIfSubgraph(Subgraph* subgraph) { in BuildIfSubgraph()
221 void SubgraphBuilder::BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs) { in BuildLessEqualCondSubgraph()
254 void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) { in BuildAccumulateLoopBodySubgraph()
300 void SubgraphBuilder::BuildPadLoopBodySubgraph(Subgraph* subgraph, in BuildPadLoopBodySubgraph()
354 void SubgraphBuilder::BuildWhileSubgraph(Subgraph* subgraph) { in BuildWhileSubgraph()
390 Subgraph* subgraph) { in BuildAssignRandomValueToVariableSubgraph()
[all …]
Dassign_variable.cc53 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
Dread_variable.cc52 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc178 explicit InterpreterInfo(Subgraph* subgraph) : subgraph_(subgraph) {} in InterpreterInfo()
206 Subgraph* subgraph_;
209 Subgraph::Subgraph(ErrorReporter* error_reporter, in Subgraph() function in tflite::Subgraph
211 std::vector<std::unique_ptr<Subgraph>>* subgraphs, in Subgraph()
241 Subgraph::~Subgraph() { in ~Subgraph()
258 void Subgraph::CleanupNode(int node_index) { in CleanupNode()
271 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
274 return static_cast<Subgraph*>(context->impl_) in ReplaceNodeSubsetsWithDelegateKernels()
373 TfLiteStatus Subgraph::ReplaceNodeSubsetsWithDelegateKernels( in ReplaceNodeSubsetsWithDelegateKernels()
440 TfLiteExternalContext* Subgraph::GetExternalContext( in GetExternalContext()
[all …]
Dsubgraph.h44 class Subgraph {
48 Subgraph(ErrorReporter* error_reporter,
50 std::vector<std::unique_ptr<Subgraph>>* subgraphs,
53 Subgraph(const Subgraph&) = delete;
56 Subgraph(Subgraph&&) = default;
57 Subgraph& operator=(const Subgraph&) = delete;
58 virtual ~Subgraph();
320 std::vector<std::unique_ptr<Subgraph>>* GetSubgraphs() { return subgraphs_; } in GetSubgraphs()
724 std::vector<std::unique_ptr<Subgraph>>* subgraphs_ = nullptr;
/external/tensorflow/tensorflow/core/graph/
Dsubgraph_test.cc107 string Subgraph(const string& fed_str, const string& fetch_str, in Subgraph() function in tensorflow::__anonf2fe6f640111::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.cc189 class Subgraph { class in tensorflow::__anondf15e6490211::Encapsulator
379 std::unordered_map<string, Subgraph> subgraphs_;
435 Node* Encapsulator::Subgraph::GetCallNode() const { return call_node_; } in GetCallNode()
437 int Encapsulator::Subgraph::GetArgIndexForEdge(const Edge* edge) const { in GetArgIndexForEdge()
441 int Encapsulator::Subgraph::GetResultIndexForEdge(const Edge* edge) const { in GetResultIndexForEdge()
445 Node* Encapsulator::Subgraph::MakeNodeImage(const Graph* graph_in, Node* node) { in MakeNodeImage()
466 Graph* Encapsulator::Subgraph::GetGraph() const { return graph_.get(); } in GetGraph()
468 Status Encapsulator::Subgraph::RecordArg( in RecordArg()
503 Status Encapsulator::Subgraph::RecordControlResult( in RecordControlResult()
512 Status Encapsulator::Subgraph::RecordResult( in RecordResult()
[all …]
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.h82 flatbuffers::FlatBufferBuilder* fbb, Subgraph* subgraph, int node_index,
88 explicit SubgraphWriter(Subgraph* subgraph) in SubgraphWriter()
120 Subgraph* subgraph, in SubgraphWriter()
185 Subgraph* subgraph_;
DREADME.md7 an `Interpreter` or `Subgraph`. Example use-cases include authoring models with
23 To generate/write a flatbuffer for a particular `Subgraph` (see
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.h89 Subgraph* subgraph);
93 Subgraph* subgraph);
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_size.cc63 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableSize()
Dhashtable_import.cc73 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableImport()
Dhashtable_find.cc80 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableFind()
Dhashtable.cc118 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtable()

12