Searched refs:IsNodeStateful (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | function_utils_test.cc | 798 EXPECT_TRUE(IsNodeStateful(lib_def, *nodeA)); in TEST() 823 EXPECT_FALSE(IsNodeStateful(lib_def, node)); in TEST() 826 EXPECT_TRUE(IsNodeStateful(lib_def, node)); in TEST() 827 EXPECT_FALSE(IsNodeStateful(lib_def, node, true)); in TEST()
|
D | function_utils.cc | 179 if (IsNodeStateful(library, node_def, skip_assert)) return true; in IsFunctionStateful() 184 bool IsNodeStateful(const FunctionLibraryDefinition& library, in IsNodeStateful() function
|
D | function_utils.h | 117 bool IsNodeStateful(const FunctionLibraryDefinition& library,
|
D | hoist_random_uniform.cc | 194 if (!function_utils::IsNodeStateful(library, node, true)) { in CanHoistRandomUniform()
|