Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc924 Node* concat_dim_node = graph->AddNode(concat_dim_def, &s); in CreateConcatNode() local
933 concat_def.add_input(absl::StrCat(concat_dim_node->name(), ":0")); in CreateConcatNode()
941 graph->AddEdge(concat_dim_node, 0, concat_node, 0); in CreateConcatNode()