Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc744 Node* split_dim_node = graph->AddNode(split_dim_def, &s); in CreateSplitNode() local
753 split_def.add_input(absl::StrCat(split_dim_node->name(), ":0")); in CreateSplitNode()
766 graph->AddEdge(split_dim_node, 0, split_node, 0); in CreateSplitNode()
773 graph->AddControlEdge(control_predecessor, split_dim_node); in CreateSplitNode()