Searched refs:argdef (Results 1 – 3 of 3) sorted by relevance
133 OpDef::ArgDef* argdef = in GraphToFunctionDef() local135 argdef->set_type(type); in GraphToFunctionDef()137 argdef->set_name(normalized); in GraphToFunctionDef()150 OpDef::ArgDef* argdef = in GraphToFunctionDef() local152 argdef->set_type(type); in GraphToFunctionDef()154 argdef->set_name(normalized); in GraphToFunctionDef()
337 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_output_arg(); in GraphToFunctionDef() local338 argdef->set_type(node->output_type(idx)); in GraphToFunctionDef()341 argdef->set_name(output_names[i]); in GraphToFunctionDef()343 argdef->set_name(node_names.GetOutputName(node->name())); in GraphToFunctionDef()351 OpDef::ArgDef* argdef = fdef->mutable_signature()->add_input_arg(); in GraphToFunctionDef() local352 argdef->set_type(node->output_type(idx)); in GraphToFunctionDef()354 argdef->set_name(input_name); in GraphToFunctionDef()
555 func isListArg(argdef *pb.OpDef_ArgDef) bool {556 return argdef.TypeListAttr != "" || argdef.NumberAttr != ""