Searched refs:input_arg_name (Results 1 – 3 of 3) sorted by relevance
102 const std::string input_arg_name = reg->op_def.input_arg(i).name(); in ShouldIgnoreInput() local103 if (input_arg_name == kSeedInputName || in ShouldIgnoreInput()104 input_arg_name == kSeed2InputName || in ShouldIgnoreInput()105 input_arg_name == kSeedGeneratorInputName) { in ShouldIgnoreInput()106 VLOG(2) << "Ignoring arg: " << input_arg_name in ShouldIgnoreInput()
389 string input_arg_name = strings::StrCat(function_node->name(), "_", i); in AddBranch() local390 input_arg->set_name(input_arg_name); in AddBranch()392 function_node->add_input(input_arg_name); in AddBranch()
123 std::string input_arg_name = in KernelArgInfo() local125 strcpy(arg_name, input_arg_name.c_str()); in KernelArgInfo()