Searched refs:keep_node_fetchable (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_while_op.cc | 68 bool keep_node_fetchable) { in Run() argument
70 graph, flib_def, keep_node_fetchable); in Run()
81 bool keep_node_fetchable);
184 bool keep_node_fetchable) in LowerWhileHelper() argument
190 keep_node_fetchable_(keep_node_fetchable), in LowerWhileHelper()
509 bool keep_node_fetchable) { in RewriteWhileNode() argument
510 VLOG(2) << "Lower While node (keep_node_fetchable=" << keep_node_fetchable in RewriteWhileNode()
532 flib_def, keep_node_fetchable)); in RewriteWhileNode()
|
| H A D | lower_case_op.cc | 42 bool keep_node_fetchable, Graph* graph);
100 bool keep_node_fetchable, Graph* graph) in CaseBuilder() argument
105 keep_node_fetchable_(keep_node_fetchable), in CaseBuilder()
270 Status RewriteCaseNode(Node* n, Graph* g, bool keep_node_fetchable) { in RewriteCaseNode() argument
271 VLOG(2) << "Lower Case node (keep_node_fetchable=" << keep_node_fetchable in RewriteCaseNode()
284 CaseBuilder cb(n, branch_fn_names, keep_node_fetchable, g); in RewriteCaseNode()
|
| H A D | lower_if_op.cc | 43 const NameAttrList& else_fn, bool keep_node_fetchable,
114 const NameAttrList& else_fn, bool keep_node_fetchable, in CondBuilder() argument
120 keep_node_fetchable_(keep_node_fetchable), in CondBuilder()
300 Status RewriteIfNode(Node* n, Graph* g, bool keep_node_fetchable) { in RewriteIfNode() argument
301 VLOG(2) << "Lower If node (keep_node_fetchable=" << keep_node_fetchable in RewriteIfNode()
313 CondBuilder cb(n, then_attr->func(), else_attr->func(), keep_node_fetchable, in RewriteIfNode()
|
| H A D | lower_if_op.h | 27 Status RewriteIfNode(Node* n, Graph* g, bool keep_node_fetchable);
|
| H A D | lower_case_op.h | 27 Status RewriteCaseNode(Node* n, Graph* g, bool keep_node_fetchable);
|
| H A D | lower_while_op.h | 31 bool keep_node_fetchable);
|