Searched refs:args_to_lift (Results 1 – 1 of 1) sorted by relevance
1905 const absl::flat_hash_set<int>& args_to_lift, in MakeIdentityNodesForArgsToLift() argument1911 if (!args_to_lift.contains(arg_index)) continue; in MakeIdentityNodesForArgsToLift()1942 const absl::flat_hash_set<int>& args_to_lift, in RemoveArgsToLiftFromFunctionBody() argument1950 if (!args_to_lift.contains(i)) { in RemoveArgsToLiftFromFunctionBody()2071 const absl::flat_hash_set<int>& args_to_lift, Graph* g, Node* while_node) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile() argument2074 if (args_to_lift.contains(e->src_output())) { in ReplaceOutputEdgesWithInputEdgeSourceForWhile()2092 const int num_args, const absl::flat_hash_set<int>& args_to_lift) { in ArgIndexMapping() argument2096 if (!args_to_lift.contains(i)) { in ArgIndexMapping()2128 TF_ASSIGN_OR_RETURN(absl::flat_hash_set<int> args_to_lift, in LiftOutsideCompilationOnlyArgsFromWhileNode()2130 if (args_to_lift.empty()) return Status::OK(); in LiftOutsideCompilationOnlyArgsFromWhileNode()[all …]