Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils_test.cc784 TEST(FunctionUtilsTest, IsFunctionStateful) { in TEST() argument
795 EXPECT_FALSE(IsFunctionStateful(lib_def, *function)); in TEST()
810 EXPECT_FALSE(IsFunctionStateful(cond_lib, *no_op_fnc)); in TEST()
811 EXPECT_FALSE(IsFunctionStateful(cond_lib, *no_op_fnc, true)); in TEST()
815 EXPECT_TRUE(IsFunctionStateful(cond_lib, *assert_func)); in TEST()
816 EXPECT_FALSE(IsFunctionStateful(cond_lib, *assert_func, true)); in TEST()
833 EXPECT_TRUE(IsFunctionStateful(cond_lib, *cond_func)); in TEST()
834 EXPECT_FALSE(IsFunctionStateful(cond_lib, *cond_func, true)); in TEST()
846 EXPECT_FALSE(IsFunctionStateful(while_lib, *while_function)); in TEST()
847 EXPECT_FALSE(IsFunctionStateful(while_lib, *while_function, true)); in TEST()
Dfunction_utils.cc174 bool IsFunctionStateful(const FunctionLibraryDefinition& library, in IsFunctionStateful() function
203 !IsFunctionStateful(library, *then_func, skip_assert)) && in IsNodeStateful()
205 !IsFunctionStateful(library, *else_func, skip_assert))) { in IsNodeStateful()
216 !IsFunctionStateful(library, *cond_func, skip_assert)) && in IsNodeStateful()
218 !IsFunctionStateful(library, *body_func, skip_assert))) { in IsNodeStateful()
Dfunction_utils.h109 bool IsFunctionStateful(const FunctionLibraryDefinition& library,
Dmap_parallelization.cc75 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
Dmap_vectorization.cc499 if (function_utils::IsFunctionStateful(function_library, *map_fn)) { in FindMapAndBatchPattern()