Searched refs:ReplaceOpType (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_rewriter_transform_test.cc | 171 void ReplaceOpType(const std::unordered_set<string>& op_name, in ReplaceOpType() function in tensorflow::graph_transforms::__anon7e4ad84b0111::FuseRemoteGraphMultipleAddOpsRewriterTest
284 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
292 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
300 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
309 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
349 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
360 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
|
| D | remote_fused_graph_execute_utils_test.cc | 153 void ReplaceOpType(const std::unordered_set<string>& op_name, in ReplaceOpType() function in tensorflow::__anon388f233f0111::FuseRemoteGraphMultipleAddOpsTest
710 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
723 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
735 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
747 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
865 ReplaceOpType({"H", "I", "J"}, "Mul"); in TEST_F()
883 ReplaceOpType({"F", "G", "H", "I", "J"}, "Mul"); in TEST_F()
|