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.cc366 absl::flat_hash_set<const OutputArgInstantiation*> remove_output_args; in RemoveFunctionOutputs() local
368 return remove_output_args.find(&output) != remove_output_args.end(); in RemoveFunctionOutputs()
376 remove_output_args.insert(&output); in RemoveFunctionOutputs()
377 } else if (!remove_output_args.empty()) { in RemoveFunctionOutputs()
379 output_mapping->push_back({i, i - remove_output_args.size()}); in RemoveFunctionOutputs()