Searched refs:concat_node (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | backports.cc | 39 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()
|
D | backports_test.cc | 75 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()
|
D | fold_old_batch_norms.cc | 232 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()
|
D | sparsify_gather_test.cc | 150 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/ |
D | layout_optimizer_test.cc | 665 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 …]
|
D | arithmetic_optimizer.cc | 1625 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()
|