Searched refs:IsFunctionStateful (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils_test.cc | 796 TEST(FunctionUtilsTest, IsFunctionStateful) { in TEST() argument
807 EXPECT_FALSE(IsFunctionStateful(lib_def, *function)); in TEST()
822 EXPECT_FALSE(IsFunctionStateful(cond_lib, *no_op_fnc)); in TEST()
823 EXPECT_FALSE(IsFunctionStateful(cond_lib, *no_op_fnc, true)); in TEST()
827 EXPECT_TRUE(IsFunctionStateful(cond_lib, *assert_func)); in TEST()
828 EXPECT_FALSE(IsFunctionStateful(cond_lib, *assert_func, true)); in TEST()
845 EXPECT_TRUE(IsFunctionStateful(cond_lib, *cond_func)); in TEST()
846 EXPECT_FALSE(IsFunctionStateful(cond_lib, *cond_func, true)); in TEST()
858 EXPECT_FALSE(IsFunctionStateful(while_lib, *while_function)); in TEST()
859 EXPECT_FALSE(IsFunctionStateful(while_lib, *while_function, true)); in TEST()
|
| D | function_utils.cc | 183 bool IsFunctionStateful(const FunctionLibraryDefinition& library, in IsFunctionStateful() function
212 !IsFunctionStateful(library, *then_func, skip_assert)) && in IsNodeStateful()
214 !IsFunctionStateful(library, *else_func, skip_assert))) { in IsNodeStateful()
225 !IsFunctionStateful(library, *cond_func, skip_assert)) && in IsNodeStateful()
227 !IsFunctionStateful(library, *body_func, skip_assert))) { in IsNodeStateful()
|
| D | function_utils.h | 113 bool IsFunctionStateful(const FunctionLibraryDefinition& library,
|
| D | map_parallelization.cc | 89 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
|
| D | noop_elimination.cc | 116 if (function_utils::IsFunctionStateful(function_library, *fdef)) return false; in IsMapIdentity()
|
| D | map_vectorization.cc | 521 if (function_utils::IsFunctionStateful(function_library, *map_fn)) { in FindMapAndBatchPattern()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 241 Status IsFunctionStateful(const FunctionLibraryDefinition& library, in IsFunctionStateful() function
421 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *then_func)); in IsNodeStateful()
424 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *else_func)); in IsNodeStateful()
435 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *cond_func)); in IsNodeStateful()
438 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *body_func)); in IsNodeStateful()
722 IsFunctionStateful(*lib_def(), *(lib_def()->Find(name)))); in CheckExternalState()
|