Home
last modified time | relevance | path

Searched refs:add_node (Results 1 – 25 of 134) sorted by relevance

123456

/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dtest_tools.cc153 (*graph.add_node()) = MakeNodeConst("node1"); in TestGraphs()
154 (*graph.add_node()) = MakeNodeSub("node2", "node3:1", "node3:0"); in TestGraphs()
155 auto node3 = graph.add_node(); in TestGraphs()
162 (*graph.add_node()) = MakeNodeConst("const1_1"); in TestGraphs()
163 (*graph.add_node()) = MakeNodeConst("const1_2"); in TestGraphs()
164 (*graph.add_node()) = MakeNodeAddN("add1", "const1_1", "const1_2"); in TestGraphs()
166 (*graph.add_node()) = MakeNodeConst("const2_1"); in TestGraphs()
167 (*graph.add_node()) = MakeNodeConst("const2_2"); in TestGraphs()
168 (*graph.add_node()) = MakeNodeConst("const2_3"); in TestGraphs()
170 auto add2 = graph.add_node(); in TestGraphs()
[all …]
Dsubgraph_test.cc41 (*graph.add_node()) = MakeNodeConst("node1"); in TEST()
42 (*graph.add_node()) = MakeNodeConst("node2"); in TEST()
84 (*graph.add_node()) = MakeNodeConst("node1"); in TEST()
85 (*graph.add_node()) = MakeNodeSub("node2", "node3:1", "node3:0"); in TEST()
86 auto node3 = graph.add_node(); in TEST()
149 (*graph.add_node()) = MakeNodeConst("node1"); in TEST()
150 (*graph.add_node()) = MakeNodeSub("node2", "node3", "node3"); in TEST()
151 (*graph.add_node()) = MakeNodeAddN("node3", "node1", "node2"); in TEST()
199 (*graph.add_node()) = MakeNodeConst("node1"); in TEST()
200 (*graph.add_node()) = MakeNodeSub("node2", "node3", "node3"); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dframe_test.cc92 *graph.add_node() = this->CreateNode("0", {}); in TYPED_TEST()
93 *graph.add_node() = this->CreateNode("1", "Enter", "while/context1", {"0"}); in TYPED_TEST()
94 *graph.add_node() = this->CreateNode("2", {"1"}); in TYPED_TEST()
95 *graph.add_node() = this->CreateNode("3", "Merge", {"2", "14"}); in TYPED_TEST()
96 *graph.add_node() = this->CreateNode("4", {"3"}); in TYPED_TEST()
97 *graph.add_node() = this->CreateNode("5", "Switch", {"4"}); in TYPED_TEST()
98 *graph.add_node() = this->CreateNode("6", {"5"}); in TYPED_TEST()
99 *graph.add_node() = this->CreateNode("7", "Enter", "while/context2", {"6"}); in TYPED_TEST()
100 *graph.add_node() = this->CreateNode("8", {"7"}); in TYPED_TEST()
101 *graph.add_node() = this->CreateNode("9", "Merge", {"8", "12"}); in TYPED_TEST()
[all …]
Dscc_test.cc74 *graph.add_node() = CreateNode("a", {"d"}); in TEST_F()
75 *graph.add_node() = CreateNode("b", {"a"}); in TEST_F()
76 *graph.add_node() = CreateNode("c", {"b"}); in TEST_F()
77 *graph.add_node() = CreateNode("d", {"c"}); in TEST_F()
80 *graph.add_node() = CreateNode("e", {}); in TEST_F()
81 *graph.add_node() = CreateNode("f", {"e"}); in TEST_F()
82 *graph.add_node() = CreateNode("g", {"f"}); in TEST_F()
83 *graph.add_node() = CreateNode("h", {"g"}); in TEST_F()
139 *graph.add_node() = CreateNode("a", {"c"}); in TEST_F()
140 *graph.add_node() = CreateNode("b", {"a", "d"}); in TEST_F()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes_test.cc41 NodeDef* add_node = graph_def.add_node(); in TestSimpleAdd() local
42 add_node->set_name("add_node"); in TestSimpleAdd()
43 add_node->set_op("Add"); in TestSimpleAdd()
44 add_node->add_input("a_node"); in TestSimpleAdd()
45 add_node->add_input("b_node"); in TestSimpleAdd()
47 NodeDef* a_node = graph_def.add_node(); in TestSimpleAdd()
51 NodeDef* b_node = graph_def.add_node(); in TestSimpleAdd()
55 NodeDef* c_node = graph_def.add_node(); in TestSimpleAdd()
73 NodeDef* add_node1 = graph_def.add_node(); in TestCommonAncestor()
79 NodeDef* add_node2 = graph_def.add_node(); in TestCommonAncestor()
[all …]
Dremove_nodes_test.cc41 NodeDef* add_node1 = graph_def.add_node(); in TestRemoveNodes()
47 NodeDef* add_node2 = graph_def.add_node(); in TestRemoveNodes()
53 NodeDef* add_node3 = graph_def.add_node(); in TestRemoveNodes()
59 NodeDef* identity_node1 = graph_def.add_node(); in TestRemoveNodes()
64 NodeDef* identity_node2 = graph_def.add_node(); in TestRemoveNodes()
69 NodeDef* identity_node3 = graph_def.add_node(); in TestRemoveNodes()
74 NodeDef* const_node1 = graph_def.add_node(); in TestRemoveNodes()
78 NodeDef* const_node2 = graph_def.add_node(); in TestRemoveNodes()
82 NodeDef* const_node3 = graph_def.add_node(); in TestRemoveNodes()
86 NodeDef* add_node4 = graph_def.add_node(); in TestRemoveNodes()
[all …]
Dsort_by_execution_order_test.cc42 NodeDef* add_node = graph_def.add_node(); in TestSimpleAdd() local
43 add_node->set_name("add_node"); in TestSimpleAdd()
44 add_node->set_op("Add"); in TestSimpleAdd()
45 add_node->add_input("a_node"); in TestSimpleAdd()
46 add_node->add_input("b_node"); in TestSimpleAdd()
48 NodeDef* b_node = graph_def.add_node(); in TestSimpleAdd()
52 NodeDef* a_node = graph_def.add_node(); in TestSimpleAdd()
69 NodeDef* negative_node = graph_def.add_node(); in TestSimpleLinear()
74 NodeDef* relu_node = graph_def.add_node(); in TestSimpleLinear()
79 NodeDef* sqrt_node = graph_def.add_node(); in TestSimpleLinear()
[all …]
Dbackports_test.cc45 NodeDef* mul_node1 = graph_def.add_node(); in TestBackportConcatV2()
51 NodeDef* add_node2 = graph_def.add_node(); in TestBackportConcatV2()
57 NodeDef* add_node3 = graph_def.add_node(); in TestBackportConcatV2()
63 NodeDef* const_node1 = graph_def.add_node(); in TestBackportConcatV2()
67 NodeDef* const_node2 = graph_def.add_node(); in TestBackportConcatV2()
71 NodeDef* const_node3 = graph_def.add_node(); in TestBackportConcatV2()
75 NodeDef* concat_node = graph_def.add_node(); in TestBackportConcatV2()
111 NodeDef* size_node = graph_def.add_node(); in TEST()
118 NodeDef* tensor_array_node = graph_def.add_node(); in TEST()
128 NodeDef* handle_output_node = graph_def.add_node(); in TEST()
[all …]
Dobfuscate_names_test.cc41 NodeDef* add_node1 = graph_def.add_node(); in TestSimpleTree()
47 NodeDef* add_node2 = graph_def.add_node(); in TestSimpleTree()
53 NodeDef* add_node3 = graph_def.add_node(); in TestSimpleTree()
59 NodeDef* const_node1 = graph_def.add_node(); in TestSimpleTree()
63 NodeDef* const_node2 = graph_def.add_node(); in TestSimpleTree()
67 NodeDef* const_node3 = graph_def.add_node(); in TestSimpleTree()
71 NodeDef* const_node4 = graph_def.add_node(); in TestSimpleTree()
94 NodeDef* const_node = graph_def.add_node(); in TestManyNodes()
113 NodeDef* const_node = graph_def.add_node(); in TestNameClashes()
Dremove_device_test.cc41 NodeDef* mul_node1 = graph_def.add_node(); in TestRemoveDevice()
48 NodeDef* add_node2 = graph_def.add_node(); in TestRemoveDevice()
55 NodeDef* add_node3 = graph_def.add_node(); in TestRemoveDevice()
61 NodeDef* const_node1 = graph_def.add_node(); in TestRemoveDevice()
65 NodeDef* const_node2 = graph_def.add_node(); in TestRemoveDevice()
69 NodeDef* const_node3 = graph_def.add_node(); in TestRemoveDevice()
73 NodeDef* add_node4 = graph_def.add_node(); in TestRemoveDevice()
Drename_op_test.cc41 NodeDef* mul_node1 = graph_def.add_node(); in TestRenameOp()
47 NodeDef* add_node2 = graph_def.add_node(); in TestRenameOp()
53 NodeDef* add_node3 = graph_def.add_node(); in TestRenameOp()
59 NodeDef* const_node1 = graph_def.add_node(); in TestRenameOp()
63 NodeDef* const_node2 = graph_def.add_node(); in TestRenameOp()
67 NodeDef* const_node3 = graph_def.add_node(); in TestRenameOp()
71 NodeDef* add_node4 = graph_def.add_node(); in TestRenameOp()
Dset_device_test.cc40 NodeDef* mul_node1 = graph_def.add_node(); in CreateDeviceGraph()
47 NodeDef* add_node2 = graph_def.add_node(); in CreateDeviceGraph()
54 NodeDef* add_node3 = graph_def.add_node(); in CreateDeviceGraph()
60 NodeDef* const_node1 = graph_def.add_node(); in CreateDeviceGraph()
64 NodeDef* const_node2 = graph_def.add_node(); in CreateDeviceGraph()
68 NodeDef* const_node3 = graph_def.add_node(); in CreateDeviceGraph()
72 NodeDef* add_node4 = graph_def.add_node(); in CreateDeviceGraph()
Dremove_attribute_test.cc41 NodeDef* mul_node1 = graph_def.add_node(); in TestRemoveAttribute()
49 NodeDef* add_node2 = graph_def.add_node(); in TestRemoveAttribute()
58 NodeDef* add_node3 = graph_def.add_node(); in TestRemoveAttribute()
64 NodeDef* const_node1 = graph_def.add_node(); in TestRemoveAttribute()
68 NodeDef* const_node2 = graph_def.add_node(); in TestRemoveAttribute()
72 NodeDef* const_node3 = graph_def.add_node(); in TestRemoveAttribute()
76 NodeDef* add_node4 = graph_def.add_node(); in TestRemoveAttribute()
Drename_node_test.cc35 NodeDef* node = in_graph.add_node(); in TEST()
39 NodeDef* node_splitter = in_graph.add_node(); in TEST()
43 NodeDef* node_adder = in_graph.add_node(); in TEST()
70 NodeDef* node = in_graph.add_node(); in TEST()
74 NodeDef* node_splitter = in_graph.add_node(); in TEST()
78 NodeDef* node_adder = in_graph.add_node(); in TEST()
Drename_attribute_test.cc41 NodeDef* mul_node1 = graph_def.add_node(); in TestRenameAttribute()
49 NodeDef* add_node2 = graph_def.add_node(); in TestRenameAttribute()
58 NodeDef* add_node3 = graph_def.add_node(); in TestRenameAttribute()
64 NodeDef* const_node1 = graph_def.add_node(); in TestRenameAttribute()
68 NodeDef* const_node2 = graph_def.add_node(); in TestRenameAttribute()
72 NodeDef* const_node3 = graph_def.add_node(); in TestRenameAttribute()
76 NodeDef* add_node4 = graph_def.add_node(); in TestRenameAttribute()
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc41 NodeDef* node = graph_def.add_node(); in GetHash()
57 NodeDef* node1 = graph_def.add_node(); in CheckEqual()
63 NodeDef* node2 = graph_def.add_node(); in CheckEqual()
161 NodeDef* output_node = g.add_node(); in TEST_F()
168 .Finalize(g.add_node())); in TEST_F()
172 .Finalize(g.add_node())); in TEST_F()
175 .Finalize(g.add_node())); in TEST_F()
178 .Finalize(g.add_node())); in TEST_F()
181 .Finalize(g.add_node())); in TEST_F()
194 NodeDef* n1 = gd.add_node(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dremove_noop_test.cc45 auto add_node = graph.NewNode(); in TEST() local
47 ASSERT_TRUE(AddOutput(&graph, add_node, &output).ok()); in TEST()
48 add_node->operation.type = ToString(OperationType::ADD); in TEST()
49 add_node->operation.attributes = ElementwiseAttributes(); in TEST()
52 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
73 auto add_node = graph.NewNode(); in TEST() local
75 ASSERT_TRUE(AddOutput(&graph, add_node, &output).ok()); in TEST()
76 add_node->operation.type = ToString(OperationType::ADD); in TEST()
79 add_node->operation.attributes = attr; in TEST()
82 ASSERT_TRUE(ConnectTwoNodes(&graph, first_node, add_node, &temp).ok()); in TEST()
[all …]
Dmerge_padding_with_test.cc136 auto add_node = graph.NewNode(); in TEST() local
138 ASSERT_TRUE(graph.AddConsumer(add_node->id, padded->id).ok()); in TEST()
139 ASSERT_TRUE(graph.AddConsumer(add_node->id, input1->id).ok()); in TEST()
140 ASSERT_TRUE(graph.SetProducer(add_node->id, output->id).ok()); in TEST()
141 add_node->operation.type = ToString(OperationType::ADD); in TEST()
142 add_node->operation.attributes = add_attr; in TEST()
153 EXPECT_EQ(add_node, graph.nodes()[0]); in TEST()
174 auto add_node = graph.NewNode(); in TEST() local
177 ASSERT_TRUE(graph.AddConsumer(add_node->id, padded->id).ok()); in TEST()
178 ASSERT_TRUE(graph.AddConsumer(add_node->id, input1->id).ok()); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc162 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst()
204 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst()
227 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertBoolTensorConst()
251 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertIntTensorConst()
275 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateIntTensorConst()
300 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertComplex64TensorConst()
323 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateMatrixShapeTensorConst()
341 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateDummyConcatDimTensorConst()
356 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in CreateReshapeShapeTensorConst()
396 tensorflow::NodeDef* conv2d_op = tensorflow_graph->add_node(); in ConvertConvOperator()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h86 *graph_def.add_node() = std::move(node); in CreateGraphDef()
97 *graph_def.add_node() = std::move(node); in CreateGraphDef()
109 *graph_def.add_node() = std::move(node); in CreateGraphDef()
132 *graph.add_node() = std::move(node); in CreateRandomGraph()
156 *graph.add_node() = std::move(input_node); in CreateFaninFanoutNodeGraph()
163 *graph.add_node() = std::move(input_node); in CreateFaninFanoutNodeGraph()
172 *graph.add_node() = std::move(output_node); in CreateFaninFanoutNodeGraph()
180 *graph.add_node() = std::move(output_node); in CreateFaninFanoutNodeGraph()
183 *graph.add_node() = std::move(node); in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc46 .Finalize(graph_def.add_node())); in TEST()
59 .Finalize(graph_def.add_node())); in TEST()
80 .Finalize(graph_def.add_node())); in TEST()
107 .Finalize(produced_graph_def.add_node())); in TEST()
116 .Finalize(expected_graph_def.add_node())); in TEST()
142 .Finalize(produced_graph_def.add_node())); in TEST()
174 .Finalize(produced_graph_def.add_node())); in TEST()
214 .Finalize(produced_graph_def.add_node())); in TEST()
229 .Finalize(expected_graph_def.add_node())); in TEST()
247 .Finalize(graph_def.add_node())); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage_test.cc164 NodeDef* add_node; in TEST_F() local
165 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
166 ASSERT_EQ(add_node->input_size(), 2); in TEST_F()
167 EXPECT_EQ(add_node->input(0), "a"); in TEST_F()
168 EXPECT_EQ(add_node->input(1), "b"); in TEST_F()
206 NodeDef* add_node; in TEST_F() local
207 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
210 NodeDef* add_node_copy = stage.AddCopyNode("Add_1", add_node); in TEST_F()
213 ASSERT_EQ(add_node->input_size(), 2); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc78 NodeDef* input = graph.add_node(); in TEST()
81 NodeDef* filter = graph.add_node(); in TEST()
85 NodeDef* output_backprop = graph.add_node(); in TEST()
89 NodeDef* input_sizes = graph.add_node(); in TEST()
94 NodeDef* filter_sizes = graph.add_node(); in TEST()
109 .Finalize(graph.add_node())); in TEST()
112 NodeDef* conv = graph.add_node(); in TEST()
120 NodeDef* conv_bp_in = graph.add_node(); in TEST()
129 NodeDef* conv_bp_filter = graph.add_node(); in TEST()
159 .Finalize(graph.add_node())); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc257 NodeDef* shp = item.graph.add_node(); in RunInfiniteTFLoop()
266 NodeDef* r = item.graph.add_node(); in RunInfiniteTFLoop()
273 NodeDef* e = item.graph.add_node(); in RunInfiniteTFLoop()
280 NodeDef* m = item.graph.add_node(); in RunInfiniteTFLoop()
288 NodeDef* t = item.graph.add_node(); in RunInfiniteTFLoop()
298 NodeDef* c = item.graph.add_node(); in RunInfiniteTFLoop()
303 NodeDef* s = item.graph.add_node(); in RunInfiniteTFLoop()
310 NodeDef* i = item.graph.add_node(); in RunInfiniteTFLoop()
316 NodeDef* n = item.graph.add_node(); in RunInfiniteTFLoop()
322 NodeDef* x = item.graph.add_node(); in RunInfiniteTFLoop()
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc356 const NodeDef* add_node = node_map.GetNode("add"); in TEST_F() local
358 ASSERT_NE(add_node, nullptr); in TEST_F()
361 EXPECT_EQ(NumNonControlInputs(*add_node), 2); in TEST_F()
362 EXPECT_EQ(NumControlInputs(*add_node), 1); in TEST_F()
364 EXPECT_EQ(NumNonControlOutputs(*add_node, node_map), 2); in TEST_F()
366 EXPECT_EQ(NumNonControlDataOutputs(*add_node, node_map), 1); in TEST_F()
369 EXPECT_TRUE(HasControlInputs(*add_node)); in TEST_F()
370 EXPECT_TRUE(HasRegularInputs(*add_node)); in TEST_F()
371 EXPECT_TRUE(HasControlOutputs(*add_node, node_map)); in TEST_F()
372 EXPECT_TRUE(HasRegularOutputs(*add_node, node_map)); in TEST_F()
[all …]

123456