Searched refs:concat_dim_def (Results 1 – 1 of 1) sorted by relevance
911 NodeDef concat_dim_def; in CreateConcatNode() local912 concat_dim_def.set_name( in CreateConcatNode()914 concat_dim_def.set_op("Const"); in CreateConcatNode()915 AddNodeAttr("dtype", DT_INT32, &concat_dim_def); in CreateConcatNode()916 concat_dim_def.set_device(std::string(device)); in CreateConcatNode()922 AddNodeAttr("value", tensor_proto, &concat_dim_def); in CreateConcatNode()924 Node* concat_dim_node = graph->AddNode(concat_dim_def, &s); in CreateConcatNode()