Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc307 absl::flat_hash_set<string>* nodes_to_delete, in RemoveShuffleDataset()
332 absl::flat_hash_set<string>* nodes_to_delete, in RemoveShuffleDatasetV2()
354 absl::flat_hash_set<string>* nodes_to_delete, in RemoveShuffleDatasetV3()
381 absl::flat_hash_set<string>* nodes_to_delete, in ProcessDatasetSourceNode()
425 absl::flat_hash_set<string>* nodes_to_delete) { in FindFuncAndTensorSliceDataset()
452 absl::flat_hash_set<string>* nodes_to_delete) { in RecursivelyHandleOp()
556 absl::flat_hash_set<string> nodes_to_delete; in ShardByFile() local
Dmap_parallelization.cc75 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dbatch_parallelization.cc78 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_and_batch_fusion.cc116 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
Dshuffle_and_repeat_fusion.cc159 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dnoop_elimination.cc145 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dfilter_with_random_uniform_fusion.cc134 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_and_filter_fusion.cc176 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dhoist_random_uniform.cc241 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
Dmap_vectorization.cc538 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc146 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
Dremapper.cc1062 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1106 std::vector<bool>* invalidated_nodes, std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1157 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1206 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1244 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1289 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1336 std::vector<bool>* invalidated_nodes, std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1378 std::vector<bool>* nodes_to_delete) { in IsCpuCompatibleDataType()
1788 std::vector<bool> nodes_to_delete(num_nodes); in IsCpuCompatibleDataType() local
Ddependency_optimizer.cc248 std::set<int>* nodes_to_delete) { in OptimizeNode()
468 std::set<int> nodes_to_delete; in OptimizeDependencies() local
Dconstant_folding.cc1647 std::vector<int> nodes_to_delete; in FoldGraph() 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.cc1491 const absl::flat_hash_set<string>& nodes_to_delete) { in CheckNodesCanBeDeleted()
1554 const absl::flat_hash_set<string>& nodes_to_delete) { in DeleteNodes()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc388 std::vector<int32> nodes_to_delete; in PostPruneTree() local
534 std::vector<int32>* nodes_to_delete, in RecursivelyDoPostPrunePreparation()