Home
last modified time | relevance | path

Searched defs:while_node (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op_test.cc68 Node* while_node; in TEST() local
190 Node* while_node; in TEST() local
295 Node* while_node; in TEST() local
418 Node* while_node; in TEST() local
533 Node* while_node; in TEST() local
H A Dcontrol_flow_deps_to_chains.cc63 NodeDef* while_node = n->mutable_def(); in Run() local
H A Dlower_functional_ops_test.cc157 Node* while_node; in TEST() local
H A Dexecutor_test.cc662 Node* while_node; in BM_WhileLoopHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc1721 Node* while_node, FunctionLibraryDefinition* fld) { in FindArgsToLiftForWhileNode()
1997 void RemoveOutputIdentityNodesForWhileV2(Graph* g, Node* while_node) { in RemoveOutputIdentityNodesForWhileV2()
2025 const absl::flat_hash_set<int>& args_to_lift, Graph* g, Node* while_node) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile()
2078 Graph* g, Node* while_node, FunctionLibraryDefinition* fld, in LiftOutsideCompilationOnlyArgsFromWhileNode()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Drearrange_function_argument_pass_test.cc154 const Node *while_node = node_name_index.at("while"); in TEST() local
H A Dcompilability_check_util.cc260 const Node& while_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableWhile()
H A Dextract_outside_compilation_pass.cc1488 Node* while_node) { in AddSendLoopPredToLoopCond()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util_test.cc407 Node* while_node = node_name_index["while"]; in TEST() local
H A Dtf2xla_util.cc402 Graph* g, Node* while_node, const FunctionLibraryDefinition* lookup_fld, in PropagateConstIntoAndAroundWhileNode()