Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc89 input_arg_expansions_.insert( in RegisterInputArgExpansion()
147 auto input_arg = input_arg_expansions_.find(node_name); in ExpandFunctionDefInput()
148 if (input_arg != input_arg_expansions_.end()) { in ExpandFunctionDefInput()
318 input_arg_expansions_(std::move(input_arg_expansions)), in GrapplerFunctionItem()
327 for (const InputArgExpansion& input_arg : input_arg_expansions_) { in GrapplerFunctionItem()
351 return input_arg_expansions_; in inputs()
355 return input_arg_expansions_[i]; in input()
359 return input_arg_expansions_.size(); in input_size()
686 auto& inputs = item->input_arg_expansions_; in ReplaceInputWithConst()
Dfunctions.h120 absl::flat_hash_map<string, InputArgExpansion> input_arg_expansions_;
203 std::vector<InputArgExpansion> input_arg_expansions_; variable