Searched defs:matched_nodes_map (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | pattern_utils_test.cc | 175 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
258 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
357 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
378 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
414 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
435 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
551 std::map<string, int> matched_nodes_map; // label to node index map in TEST_F() local
|
| H A D | pattern_utils.cc | 171 MutableNodeView* node_view, std::map<string, int>* matched_nodes_map, in GetMatchedNodes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | remapper.cc | 1119 std::map<string, int>* matched_nodes_map, in FindMatMulBiasAddAndGelu()
1315 std::map<string, int>* matched_nodes_map, in FindMulAndMaximum()
1401 std::map<string, int>* matched_nodes_map, in FindSigmoidAndMul()
1449 std::map<string, int>* matched_nodes_map, in FindMklLayerNorm()
1971 std::map<string, int>* matched_nodes_map, in FindFusedBatchMatMul()
2598 RemapperContext* ctx, const std::map<string, int>& matched_nodes_map, in AddFusedMatMulBiasAddAndGelu()
2641 const std::map<string, int>& matched_nodes_map, in AddMklLayerNorm()
2677 RemapperContext* ctx, const std::map<string, int>& matched_nodes_map, in ReplaceMulMaximumWithLeakyRelu()
2733 RemapperContext* ctx, const std::map<string, int>& matched_nodes_map, in ReplaceSigmoidMulWithSwish()
3140 const std::map<string, int>& matched_nodes_map, in AddFusedBatchMatMul()
[all …]
|