Searched refs:size_def (Results 1 – 1 of 1) sorted by relevance
979 NodeDef size_def; in CreateSliceNode() local980 size_def.set_name( in CreateSliceNode()982 size_def.set_op("Const"); in CreateSliceNode()983 AddNodeAttr("dtype", DT_INT32, &size_def); in CreateSliceNode()984 size_def.set_device(std::string(device)); in CreateSliceNode()992 AddNodeAttr("value", sizes_tensor_proto, &size_def); in CreateSliceNode()993 Node* size_node = graph->AddNode(size_def, &s); in CreateSliceNode()