Searched refs:split_def (Results 1 – 1 of 1) sorted by relevance
747 NodeDef split_def; in CreateSplitNode() local748 split_def.set_name(graph->NewName(absl::StrCat(name_prefix, "/split"))); in CreateSplitNode()749 split_def.set_op("Split"); in CreateSplitNode()750 split_def.set_device(input_assigned_device); in CreateSplitNode()751 AddNodeAttr("num_split", num_splits, &split_def); in CreateSplitNode()752 AddNodeAttr("T", dtype, &split_def); in CreateSplitNode()753 split_def.add_input(absl::StrCat(split_dim_node->name(), ":0")); in CreateSplitNode()754 split_def.add_input(absl::StrCat(to_split_node->name(), ":", to_split_index)); in CreateSplitNode()755 Node* split_node = graph->AddNode(split_def, &s); in CreateSplitNode()