Home
last modified time | relevance | path

Searched refs:concat_node (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports.cc39 NodeDef concat_node = concat_v2_node; in BackportConcatV2Transform() local
40 concat_node.set_op("Concat"); in BackportConcatV2Transform()
44 concat_node.mutable_input()->Clear(); in BackportConcatV2Transform()
47 concat_node.add_input(dim_input); in BackportConcatV2Transform()
49 concat_node.add_input(concat_v2_node.input(i)); in BackportConcatV2Transform()
52 concat_node.mutable_attr()->erase("Tidx"); in BackportConcatV2Transform()
53 new_nodes->push_back(concat_node); in BackportConcatV2Transform()
Dbackports_test.cc75 NodeDef* concat_node = graph_def.add_node(); in TestBackportConcatV2() local
76 concat_node->set_name("concat_node"); in TestBackportConcatV2()
77 concat_node->set_op("ConcatV2"); in TestBackportConcatV2()
78 concat_node->add_input("const_node1"); in TestBackportConcatV2()
79 concat_node->add_input("const_node2"); in TestBackportConcatV2()
80 concat_node->add_input("const_node3"); in TestBackportConcatV2()
81 SetNodeAttr("Tidx", DT_INT32, concat_node); in TestBackportConcatV2()
Dfold_old_batch_norms.cc232 NodeDef concat_node = concat_node_match.node; in FuseBatchNormWithConvConcat() local
233 CHECK_EQ("ConcatV2", concat_node.op()); in FuseBatchNormWithConvConcat()
260 const string concat0_output_name = concat_node.name() + "_bn_in0"; in FuseBatchNormWithConvConcat()
266 const string concat1_output_name = concat_node.name() + "_bn_in1"; in FuseBatchNormWithConvConcat()
275 concat_node.set_name(batch_norm_node.name()); in FuseBatchNormWithConvConcat()
276 concat_node.set_input(0, concat0_output_name); in FuseBatchNormWithConvConcat()
277 concat_node.set_input(1, concat1_output_name); in FuseBatchNormWithConvConcat()
278 new_nodes->push_back(concat_node); in FuseBatchNormWithConvConcat()
Dsparsify_gather_test.cc150 NodeDef* concat_node = nullptr; in TestSinglePartition() local
152 concat_node = CreateNode( in TestSinglePartition()
155 SetNodeAttr("N", 2, concat_node); in TestSinglePartition()
159 concat_node = CreateNode("concat/node", "ConcatV2", in TestSinglePartition()
163 SetNodeAttr("N", 3, concat_node); in TestSinglePartition()
383 NodeDef* concat_node = CreateNode( in TestMultiPartition() local
386 SetNodeAttr("N", 2, concat_node); in TestMultiPartition()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc665 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
666 EXPECT_EQ(concat_node->input(0), "split:1"); in TEST_F()
667 EXPECT_EQ(concat_node->input(1), "split:1"); in TEST_F()
668 EXPECT_EQ(concat_node->input(2), "split:1"); in TEST_F()
669 EXPECT_EQ(concat_node->input(3), "concat-3-LayoutOptimizer"); in TEST_F()
687 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
688 EXPECT_EQ(concat_node->input(0), "split"); in TEST_F()
689 EXPECT_EQ(concat_node->input(1), "split:1"); in TEST_F()
690 EXPECT_EQ(concat_node->input(2), "concat-2-LayoutOptimizer"); in TEST_F()
709 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
[all …]
Darithmetic_optimizer.cc1625 NodeDef* concat_node) { in HoistChainForConcat() argument
1626 const string& concat_name = concat_node->name(); in HoistChainForConcat()
1627 const int first_input = concat_node->op() == "Concat" ? 1 : 0; in HoistChainForConcat()
1632 CHECK_LT(concat_port, concat_node->input_size()); in HoistChainForConcat()
1633 const string concat_input = concat_node->input(concat_port); in HoistChainForConcat()
1636 concat_node->set_input(concat_port, tail_input); in HoistChainForConcat()
1642 UpdateConsumers(concat_node, concat_input); in HoistChainForConcat()