Searched refs:concat_def (Results 1 – 1 of 1) sorted by relevance
928 NodeDef concat_def; in CreateConcatNode() local929 concat_def.set_name(graph->NewName(absl::StrCat(name_prefix, "/concat"))); in CreateConcatNode()930 concat_def.set_op("Concat"); in CreateConcatNode()931 AddNodeAttr("N", num_splits, &concat_def); in CreateConcatNode()932 AddNodeAttr("T", dtype, &concat_def); in CreateConcatNode()933 concat_def.add_input(absl::StrCat(concat_dim_node->name(), ":0")); in CreateConcatNode()934 concat_def.set_device(std::string(device)); in CreateConcatNode()936 concat_def.add_input(absl::StrCat(i.node->name(), ":", i.index)); in CreateConcatNode()938 Node* concat_node = graph->AddNode(concat_def, &s); in CreateConcatNode()