Searched refs:IsNodeStateful (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | serialization_utils.cc | 36 if (!IsNodeStateful(lib_def, node).ok()) { in FindStatefulOps()
45 if (!IsNodeStateful(lib_def, node).ok()) { in FindStatefulOps()
|
| D | captured_function.h | 62 Status IsNodeStateful(const FunctionLibraryDefinition& library,
|
| D | captured_function.cc | 248 TF_RETURN_IF_ERROR(IsNodeStateful(library, node_def)); in IsFunctionStateful()
403 Status IsNodeStateful(const FunctionLibraryDefinition& library, in IsNodeStateful() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils_test.cc | 810 EXPECT_TRUE(IsNodeStateful(lib_def, *nodeA)); in TEST()
835 EXPECT_FALSE(IsNodeStateful(lib_def, node)); in TEST()
838 EXPECT_TRUE(IsNodeStateful(lib_def, node)); in TEST()
839 EXPECT_FALSE(IsNodeStateful(lib_def, node, true)); in TEST()
|
| D | function_utils.cc | 188 if (IsNodeStateful(library, node_def, skip_assert)) return true; in IsFunctionStateful()
193 bool IsNodeStateful(const FunctionLibraryDefinition& library, in IsNodeStateful() function
|
| D | function_utils.h | 121 bool IsNodeStateful(const FunctionLibraryDefinition& library,
|
| D | hoist_random_uniform.cc | 204 if (!function_utils::IsNodeStateful(library, node, true)) { in CanHoistRandomUniform()
|