| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A 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
|
| H A D | filter_parallelization.cc | 73 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | batch_parallelization.cc | 75 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | map_parallelization.cc | 74 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | map_and_batch_fusion.cc | 114 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | filter_fusion.cc | 72 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | shuffle_and_repeat_fusion.cc | 162 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | map_fusion.cc | 91 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | noop_elimination.cc | 155 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | map_and_filter_fusion.cc | 172 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| H A D | make_deterministic.cc | 676 absl::flat_hash_set<string> nodes_to_delete; in OptimizeAndCollectStats() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A 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 …]
|
| H A 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
|
| H A D | dependency_optimizer.cc | 256 std::set<int>* nodes_to_delete) { in OptimizeNode()
476 std::set<int> nodes_to_delete; in OptimizeDependencies() local
|
| H A D | constant_folding.cc | 1676 std::vector<int> nodes_to_delete; in FoldGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A 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()
|
| H A 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()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | finalize.cpp | 51 std::vector<Node*> nodes_to_delete; in removePackedParamInsertionAndFPWeightsSetAttr() local
81 std::vector<Node*> nodes_to_delete; in removeObserverCallMethods() local
|
| H A D | register_packed_params.cpp | 62 std::unordered_set<Node*> nodes_to_delete; in RegisterPrePackParams() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | prepack_folding.cpp | 19 std::unordered_set<Node*> nodes_to_delete; in PrePackingOpsFolder() local
|
| H A D | mkldnn_rewrite.cpp | 163 std::unordered_set<Node*> nodes_to_delete; in PrePackingOpsFolder() local
|