Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports.cc38 NodeDef concat_node = concat_v2_node; in BackportConcatV2Transform() local
39 concat_node.set_op("Concat"); in BackportConcatV2Transform()
43 concat_node.mutable_input()->Clear(); in BackportConcatV2Transform()
46 concat_node.add_input(dim_input); in BackportConcatV2Transform()
48 concat_node.add_input(concat_v2_node.input(i)); in BackportConcatV2Transform()
51 concat_node.mutable_attr()->erase("Tidx"); in BackportConcatV2Transform()
52 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.cc231 NodeDef concat_node = concat_node_match.node; in FuseBatchNormWithConvConcat() local
232 CHECK_EQ("ConcatV2", concat_node.op()); in FuseBatchNormWithConvConcat()
259 const string concat0_output_name = concat_node.name() + "_bn_in0"; in FuseBatchNormWithConvConcat()
265 const string concat1_output_name = concat_node.name() + "_bn_in1"; in FuseBatchNormWithConvConcat()
274 concat_node.set_name(batch_norm_node.name()); in FuseBatchNormWithConvConcat()
275 concat_node.set_input(0, concat0_output_name); in FuseBatchNormWithConvConcat()
276 concat_node.set_input(1, concat1_output_name); in FuseBatchNormWithConvConcat()
277 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/lite/delegates/gpu/common/transformations/
Dmake_padding_test.cc40 auto concat_node = graph.NewNode(); in TEST() local
41 ASSERT_TRUE(graph.AddConsumer(concat_node->id, input->id).ok()); in TEST()
42 concat_node->operation.type = ToString(OperationType::CONCAT); in TEST()
45 concat_node->operation.attributes = attr; in TEST()
48 ASSERT_TRUE(AddOutput(&graph, concat_node, &output).ok()); in TEST()
61 ConnectTwoNodes(&graph, const_node, concat_node, &const_link).ok()); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc685 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
686 EXPECT_EQ(concat_node->input(0), "split:1"); in TEST_F()
687 EXPECT_EQ(concat_node->input(1), "split:1"); in TEST_F()
688 EXPECT_EQ(concat_node->input(2), "split:1"); in TEST_F()
689 EXPECT_EQ(concat_node->input(3), "concat-3-LayoutOptimizer"); in TEST_F()
707 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
708 EXPECT_EQ(concat_node->input(0), "split"); in TEST_F()
709 EXPECT_EQ(concat_node->input(1), "split:1"); in TEST_F()
710 EXPECT_EQ(concat_node->input(2), "concat-2-LayoutOptimizer"); in TEST_F()
729 auto concat_node = node_map.GetNode("concat"); in TEST_F() local
[all …]
Darithmetic_optimizer.cc1704 NodeDef* concat_node) { in HoistChainForConcat() argument
1705 const string& concat_name = concat_node->name(); in HoistChainForConcat()
1706 const int first_input = concat_node->op() == "Concat" ? 1 : 0; in HoistChainForConcat()
1711 CHECK_LT(concat_port, concat_node->input_size()); in HoistChainForConcat()
1712 const string concat_input = concat_node->input(concat_port); in HoistChainForConcat()
1715 concat_node->set_input(concat_port, tail_input); in HoistChainForConcat()
1721 UpdateConsumers(concat_node, concat_input); in HoistChainForConcat()
/external/webrtc/third_party/abseil-cpp/absl/strings/
Dcord.cc1133 CordRepConcat* concat_node = node->concat(); in Build() local
1134 if (concat_node->depth() >= kMinLengthSize || in Build()
1135 concat_node->length < min_length[concat_node->depth()]) { in Build()
1136 pending.push_back(concat_node->right); in Build()
1137 pending.push_back(concat_node->left); in Build()
1139 if (concat_node->refcount.IsOne()) { in Build()
1140 concat_node->left = concat_freelist_; in Build()
1141 concat_freelist_ = concat_node; in Build()
1143 Ref(concat_node->right); in Build()
1144 Ref(concat_node->left); in Build()
[all …]
/external/libtextclassifier/abseil-cpp/absl/strings/
Dcord.cc1110 CordRepConcat* concat_node = node->concat(); in Build() local
1111 if (concat_node->depth() >= kMinLengthSize || in Build()
1112 concat_node->length < min_length[concat_node->depth()]) { in Build()
1113 pending.push_back(concat_node->right); in Build()
1114 pending.push_back(concat_node->left); in Build()
1116 if (concat_node->refcount.IsOne()) { in Build()
1117 concat_node->left = concat_freelist_; in Build()
1118 concat_freelist_ = concat_node; in Build()
1120 Ref(concat_node->right); in Build()
1121 Ref(concat_node->left); in Build()
[all …]
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/strings/
Dcord.cc1110 CordRepConcat* concat_node = node->concat(); in Build() local
1111 if (concat_node->depth() >= kMinLengthSize || in Build()
1112 concat_node->length < min_length[concat_node->depth()]) { in Build()
1113 pending.push_back(concat_node->right); in Build()
1114 pending.push_back(concat_node->left); in Build()
1116 if (concat_node->refcount.IsOne()) { in Build()
1117 concat_node->left = concat_freelist_; in Build()
1118 concat_freelist_ = concat_node; in Build()
1120 Ref(concat_node->right); in Build()
1121 Ref(concat_node->left); in Build()
[all …]
/external/abseil-cpp/absl/strings/
Dcord.cc1133 CordRepConcat* concat_node = node->concat(); in Build() local
1134 if (concat_node->depth() >= kMinLengthSize || in Build()
1135 concat_node->length < min_length[concat_node->depth()]) { in Build()
1136 pending.push_back(concat_node->right); in Build()
1137 pending.push_back(concat_node->left); in Build()
1139 if (concat_node->refcount.IsOne()) { in Build()
1140 concat_node->left = concat_freelist_; in Build()
1141 concat_freelist_ = concat_node; in Build()
1143 Ref(concat_node->right); in Build()
1144 Ref(concat_node->left); in Build()
[all …]
/external/openscreen/third_party/abseil/src/absl/strings/
Dcord.cc1107 CordRepConcat* concat_node = node->concat(); in Build() local
1108 if (concat_node->depth() >= kMinLengthSize || in Build()
1109 concat_node->length < min_length[concat_node->depth()]) { in Build()
1110 pending.push_back(concat_node->right); in Build()
1111 pending.push_back(concat_node->left); in Build()
1113 if (concat_node->refcount.IsOne()) { in Build()
1114 concat_node->left = concat_freelist_; in Build()
1115 concat_freelist_ = concat_node; in Build()
1117 Ref(concat_node->right); in Build()
1118 Ref(concat_node->left); in Build()
[all …]
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc938 Node* concat_node = graph->AddNode(concat_def, &s); in CreateConcatNode() local
941 graph->AddEdge(concat_dim_node, 0, concat_node, 0); in CreateConcatNode()
947 graph->AddEdge(i.node, i.index, concat_node, dst_input); in CreateConcatNode()
950 return concat_node; in CreateConcatNode()
956 Node* concat_node, in CreateSliceNode() argument
963 graph->NewName(absl::StrCat(concat_node->name(), "/slice_begin"))); in CreateSliceNode()
981 graph->NewName(absl::StrCat(concat_node->name(), "/slice_size"))); in CreateSliceNode()
999 graph->NewName(absl::StrCat(concat_node->name(), "/slice"))); in CreateSliceNode()
1004 slice_def.add_input(absl::StrCat(concat_node->name(), ":", concat_out_index)); in CreateSliceNode()
1010 graph->AddEdge(concat_node, concat_out_index, slice_node, 0); in CreateSliceNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc960 Node* concat_node = graph->NewNode(); in ParseBasic() local
961 concat_node->operation.type = ToString(OperationType::CONCAT); in ParseBasic()
964 concat_node->operation.attributes = concat_attr; in ParseBasic()
987 RETURN_IF_ERROR(reader->AddInput(concat_node, 0)); // input in ParseBasic()
988 RETURN_IF_ERROR(reader->AddInput(concat_node, 1)); // prev_activ in ParseBasic()
989 RETURN_IF_ERROR(graph->SetProducer(concat_node->id, concat_temp->id)); in ParseBasic()