Searched refs:split_dim_def (Results 1 – 1 of 1) sorted by relevance
731 NodeDef split_dim_def; in CreateSplitNode() local732 split_dim_def.set_name( in CreateSplitNode()734 split_dim_def.set_op("Const"); in CreateSplitNode()735 split_dim_def.set_device(input_assigned_device); in CreateSplitNode()736 AddNodeAttr("dtype", DT_INT32, &split_dim_def); in CreateSplitNode()742 AddNodeAttr("value", tensor_proto, &split_dim_def); in CreateSplitNode()744 Node* split_dim_node = graph->AddNode(split_dim_def, &s); in CreateSplitNode()