/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | auto_shard.cc | 307 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
|
D | map_parallelization.cc | 75 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | batch_parallelization.cc | 78 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | map_and_batch_fusion.cc | 116 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | filter_fusion.cc | 70 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | map_fusion.cc | 90 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | shuffle_and_repeat_fusion.cc | 159 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | noop_elimination.cc | 145 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | filter_with_random_uniform_fusion.cc | 134 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | map_and_filter_fusion.cc | 176 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | hoist_random_uniform.cc | 241 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
D | map_vectorization.cc | 538 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
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 | remapper.cc | 1062 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
|
D | dependency_optimizer.cc | 248 std::set<int>* nodes_to_delete) { in OptimizeNode() 468 std::set<int> nodes_to_delete; in OptimizeDependencies() local
|
D | constant_folding.cc | 1647 std::vector<int> nodes_to_delete; in FoldGraph() local
|
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.cc | 381 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()
|
D | mutable_graph_view.cc | 1491 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/ |
D | resources.cc | 388 std::vector<int32> nodes_to_delete; in PostPruneTree() local 534 std::vector<int32>* nodes_to_delete, in RecursivelyDoPostPrunePreparation()
|