Searched defs:nodes_to_delete (Results  1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ | 
| D | auto_shard.cc | 366                             absl::flat_hash_set<string>* nodes_to_delete,  in RemoveShuffleDataset()391                               absl::flat_hash_set<string>* nodes_to_delete,  in RemoveShuffleDatasetV2()
 413                               absl::flat_hash_set<string>* nodes_to_delete,  in RemoveShuffleDatasetV3()
 440                                 absl::flat_hash_set<string>* nodes_to_delete,  in ProcessDatasetSourceNode()
 484     absl::flat_hash_set<string>* nodes_to_delete) {  in FindFuncAndTensorSliceDataset()
 545                            absl::flat_hash_set<string>* nodes_to_delete) {  in RecursivelyHandleOp()
 649   absl::flat_hash_set<string> nodes_to_delete;  in ShardByFile()  local
 
 | 
| D | filter_parallelization.cc | 73   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | batch_parallelization.cc | 75   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | map_parallelization.cc | 74   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | map_and_batch_fusion.cc | 114   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | filter_fusion.cc | 72   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | shuffle_and_repeat_fusion.cc | 162   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | map_fusion.cc | 91   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | noop_elimination.cc | 155   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | map_and_filter_fusion.cc | 172   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| D | make_deterministic.cc | 676   absl::flat_hash_set<string> nodes_to_delete;  in OptimizeAndCollectStats()  local
 | 
| /external/tensorflow/tensorflow/core/grappler/ | 
| D | utils.cc | 366 void EraseNodesFromGraphImpl(const UniqueContainer& nodes_to_delete,  in EraseNodesFromGraphImpl()388 void EraseNodesFromGraph(const std::set<int>& nodes_to_delete,  in EraseNodesFromGraph()
 393 void EraseNodesFromGraph(std::vector<int>&& nodes_to_delete, GraphDef* graph) {  in EraseNodesFromGraph()
 398 void EraseNodesFromGraph(const std::set<string>& nodes_to_delete,  in EraseNodesFromGraph()
 
 | 
| D | mutable_graph_view.cc | 1507     const absl::flat_hash_set<string>& nodes_to_delete) {  in CheckNodesCanBeDeleted()1570     const absl::flat_hash_set<string>& nodes_to_delete) {  in DeleteNodes()
 
 | 
| /external/tensorflow/tensorflow/core/grappler/optimizers/ | 
| D | remapper.cc | 2210                                std::vector<bool>* nodes_to_delete) {  in AddFusedContractionNode()2255     std::vector<bool>* invalidated_nodes, std::vector<bool>* nodes_to_delete) {  in AddFusedContractionNode()
 2309                         std::vector<bool>* nodes_to_delete) {  in AddFusedConvNode()
 2363                           std::vector<bool>* nodes_to_delete) {  in AddFusedConv2DNode()
 2401                           std::vector<bool>* nodes_to_delete) {  in AddFusedConv2DNode()
 2445                                std::vector<bool>* nodes_to_delete) {  in AddFusedContractionNode()
 2496                           std::vector<bool>* nodes_to_delete) {  in AddFusedConv3DNode()
 2553     std::vector<bool>* invalidated_nodes, std::vector<bool>* nodes_to_delete) {  in AddFusedContractionNode()
 2600     std::vector<bool>* invalidated_nodes, std::vector<bool>* nodes_to_delete,  in AddFusedMatMulBiasAddAndGelu()
 2644                        std::vector<bool>* nodes_to_delete) {  in AddMklLayerNorm()
 [all …]
 
 | 
| D | model_pruner.cc | 146     const absl::flat_hash_set<const NodeDef*>& nodes_to_delete,  in ForwardInputsInternal()180                    const absl::flat_hash_set<const NodeDef*>& nodes_to_delete,  in ForwardInputs()
 461   absl::flat_hash_set<const NodeDef*> nodes_to_delete;  in Optimize()  local
 
 | 
| D | dependency_optimizer.cc | 256                                        std::set<int>* nodes_to_delete) {  in OptimizeNode()476   std::set<int> nodes_to_delete;  in OptimizeDependencies()  local
 
 | 
| D | constant_folding.cc | 1676   std::vector<int> nodes_to_delete;  in FoldGraph()  local
 |