Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc752 absl::flat_hash_set<const OutputArgExpansion*> remove_output_args; in RemoveFunctionOutputs() local
754 return remove_output_args.find(&output) != remove_output_args.end(); in RemoveFunctionOutputs()
762 remove_output_args.insert(&output); in RemoveFunctionOutputs()
763 } else if (!remove_output_args.empty()) { in RemoveFunctionOutputs()
765 output_mapping->push_back({i, i - remove_output_args.size()}); in RemoveFunctionOutputs()