Home
last modified time | relevance | path

Searched refs:input_arg_expansion (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc80 InputArgExpansion input_arg_expansion) { in RegisterInputArgExpansion() argument
81 string input_name = input_arg_expansion.input_name; in RegisterInputArgExpansion()
82 const auto& placeholders = input_arg_expansion.placeholders; in RegisterInputArgExpansion()
85 const string& placeholder = input_arg_expansion.placeholders[i]; in RegisterInputArgExpansion()
90 {std::move(input_name), std::move(input_arg_expansion)}); in RegisterInputArgExpansion()
149 const InputArgExpansion& input_arg_expansion = input_arg->second; in ExpandFunctionDefInput() local
150 const auto& placeholders = input_arg_expansion.placeholders; in ExpandFunctionDefInput()
159 if (position > input_arg_expansion.placeholders.size() - 1) { in ExpandFunctionDefInput()
164 graph_def_inputs->push_back(input_arg_expansion.placeholders[position]); in ExpandFunctionDefInput()
690 InputArgExpansion* input_arg_expansion = nullptr; in ReplaceInputWithConst() local
[all …]
Dfunctions.h90 void RegisterInputArgExpansion(InputArgExpansion input_arg_expansion);