Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h103 absl::flat_hash_set<string>* nodes_to_not_simplify);
109 absl::flat_hash_set<string>* nodes_to_not_simplify);
Dconstant_folding.cc1435 GraphDef* output, absl::flat_hash_set<string>* nodes_to_not_simplify) { in FoldGraph() argument
1465 nodes_to_not_simplify->emplace(node->name()); in FoldGraph()
1752 absl::flat_hash_set<string>* nodes_to_not_simplify) { in SimplifyGraph() argument
1757 if (nodes_to_not_simplify->find(node->name()) == in SimplifyGraph()
1758 nodes_to_not_simplify->end()) { in SimplifyGraph()
1760 nodes_to_not_simplify->insert(node->name()); in SimplifyGraph()
3399 absl::flat_hash_set<string> nodes_to_not_simplify; in RunOptimizationPass() local
3400 TF_RETURN_IF_ERROR(FoldGraph(optimized_graph, &nodes_to_not_simplify)); in RunOptimizationPass()
3403 &properties, &nodes_to_not_simplify)); in RunOptimizationPass()