Searched refs:GetOpTypeMatches (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils_test.cc | 305 TF_ASSERT_OK(matcher.GetOpTypeMatches({"Const"}, &const_matches)); in TestGetOpTypeMatches()
314 TF_ASSERT_OK(matcher.GetOpTypeMatches({"Add"}, &add_matches)); in TestGetOpTypeMatches()
320 TF_ASSERT_OK(matcher.GetOpTypeMatches({"Add", {{"Const"}, {"Const"}}}, in TestGetOpTypeMatches()
333 TF_ASSERT_OK(matcher.GetOpTypeMatches({"NoSuch"}, &no_such_matches)); in TestGetOpTypeMatches()
337 TF_ASSERT_OK(matcher.GetOpTypeMatches( in TestGetOpTypeMatches()
359 matcher.GetOpTypeMatches({"*", {{"*"}, {"*"}}}, &wildcard_matches)); in TestGetOpTypeMatches()
368 TF_ASSERT_OK(matcher.GetOpTypeMatches({"Add|Mul"}, &or_matches)); in TestGetOpTypeMatches()
398 TF_ASSERT_OK(matcher.GetOpTypeMatches({"Add", {{"Const"}, {"Const"}}}, in TestGetOpTypeMatchesDAG()
|
| D | transform_utils.h | 170 Status GetOpTypeMatches(const OpTypePattern& pattern,
|
| D | transform_utils.cc | 291 Status GraphMatcher::GetOpTypeMatches(const OpTypePattern& pattern, in GetOpTypeMatches() function in tensorflow::graph_transforms::GraphMatcher
372 TF_RETURN_IF_ERROR(matcher.GetOpTypeMatches(pattern, &matches)); in ReplaceMatchingOpTypes()
|