Home
last modified time | relevance | path

Searched refs:ContainsFunctionNodeWithOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils_test.cc171 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
251 function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
256 ASSERT_TRUE(function_utils::ContainsFunctionNodeWithOp("Cast", *vectorized)); in TEST()
325 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
406 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
561 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
648 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
727 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
795 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
877 !function_utils::ContainsFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
[all …]
Dfunction_utils_test.cc70 TEST(FunctionUtilsTest, ContainsFunctionNodeWithOp) { in TEST() argument
72 EXPECT_FALSE(ContainsFunctionNodeWithOp("weird_op_that_should_not_be_there", in TEST()
74 EXPECT_TRUE(ContainsFunctionNodeWithOp("Mul", function)); in TEST()
818 EXPECT_TRUE(ContainsFunctionNodeWithOp("Const", *assert_func)); in TEST()
819 EXPECT_TRUE(ContainsFunctionNodeWithOp("Assert", *assert_func)); in TEST()
Dfunction_utils.h77 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function);
Dfusion_utils_test.cc114 function_utils::ContainsFunctionNodeWithOp("Equal", *fused_function)); in TEST()
Dfunction_utils.cc130 bool ContainsFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { in ContainsFunctionNodeWithOp() function