MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsFunctionStateful

Function IsFunctionStateful

tensorflow/core/kernels/data/captured_function.cc:221–231  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

219 const NodeDef& node);
220
221Status IsFunctionStateful(const FunctionLibraryDefinition& library,
222 const FunctionDef& function_def) {
223 if (!function_def.signature().is_stateful()) {
224 return Status::OK();
225 }
226
227 for (const NodeDef& node_def : function_def.node_def()) {
228 TF_RETURN_IF_ERROR(IsNodeStateful(library, node_def));
229 }
230 return Status::OK();
231}
232
233// Returns whether an op has been whitelisted as stateless. Uses a heuristic to
234// whitelist source dataset ops which have been marked stateful due to

Callers 2

IsNodeStatefulFunction · 0.70
CheckExternalStateMethod · 0.70

Calls 4

signatureMethod · 0.80
IsNodeStatefulFunction · 0.70
is_statefulMethod · 0.45
node_defMethod · 0.45

Tested by

no test coverage detected