Home
last modified time | relevance | path

Searched defs:nodes_to_delete (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc161 absl::flat_hash_set<string>* nodes_to_delete) { in RemoveShuffleDataset()
179 absl::flat_hash_set<string>* nodes_to_delete) { in RecursivelyHandleOp()
234 absl::flat_hash_set<string> nodes_to_delete; in OptimizeGraph() local
Dmake_numa_aware.cc47 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dshuffle_and_repeat_fusion.cc42 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_and_batch_fusion.cc108 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_parallelization.cc61 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dnoop_elimination.cc80 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dfilter_fusion.cc70 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_fusion.cc90 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_and_filter_fusion.cc107 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dhoist_random_uniform.cc231 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_vectorization.cc516 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc381 void EraseNodesFromGraphImpl(const UniqueContainer& nodes_to_delete, in EraseNodesFromGraphImpl()
403 void EraseNodesFromGraph(const std::set<int>& nodes_to_delete, in EraseNodesFromGraph()
408 void EraseNodesFromGraph(std::vector<int>&& nodes_to_delete, GraphDef* graph) { in EraseNodesFromGraph()
413 void EraseNodesFromGraph(const std::set<string>& nodes_to_delete, in EraseNodesFromGraph()
Dmutable_graph_view.cc1488 const absl::flat_hash_set<string>& nodes_to_delete) { in CheckNodesCanBeDeleted()
1551 const absl::flat_hash_set<string>& nodes_to_delete) { in DeleteNodes()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc128 const absl::flat_hash_set<const NodeDef*>& nodes_to_delete, in ForwardInputsInternal()
162 const absl::flat_hash_set<const NodeDef*>& nodes_to_delete, in ForwardInputs()
458 absl::flat_hash_set<const NodeDef*> nodes_to_delete; in Optimize() local
Ddependency_optimizer.cc226 std::set<int>* nodes_to_delete) { in OptimizeNode()
445 std::set<int> nodes_to_delete; in OptimizeDependencies() local
Dconstant_folding.cc1477 std::vector<int> nodes_to_delete; in FoldGraph() local
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc313 std::vector<int32> nodes_to_delete; in PostPruneTree() local
444 std::vector<int32>* nodes_to_delete, in RecursivelyDoPostPrunePreparation()