Searched defs:arg_node (Results 1 – 10 of 10) sorted by relevance
80 Node* arg_node = nullptr; in AddArgNodes() local
190 Node* arg_node = arg_nodes[arg_index]; in ReplaceArgUsageWithConstNode() local
671 Node* arg_node = xla_graph->AddNode(arg_def, &s); in MoveHeadOutsideCompilationToHost() local1787 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1819 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1830 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1860 const Node* arg_node = fbody.arg_nodes[i]; in FindArgsToLiftForCallNode() local1948 Node* arg_node = fbody->arg_nodes[i]; in RemoveArgsToLiftFromFunctionBody() local
163 const Node* arg_node = arg_index_node_pair.second; in Initialize() local
518 struct deref_node *arg_node = get_deref_node(arg_deref, state); in lower_copies_to_load_store() local
523 Node* arg_node = function_body.graph->AddNode(arg_def, &s); in AddOutsideCompilationInputArgToFunctionBody() local532 Node* arg_node) { in AddMatchingRetvalNode()550 Node* arg_node) { in ReplaceLiftedArgNodePlaceholderWithArg()
221 const Edge* FindHostToDeviceEdge(Node* arg_node) { in FindHostToDeviceEdge()557 Node* arg_node = input_edge->src(); in FindTpuReplicatedInputAndXlaSharding() local1985 Node* arg_node; in ShardInputsWithXlaSharding() local
477 static void write_call_args_tree(FILE *out, struct cil_tree_node *arg_node) in write_call_args_tree()
2731 struct cil_tree_node *arg_node = NULL; in cil_build_call_args() local
1589 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local2519 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local2605 auto& arg_node = arg_node_and_idx.value(); in GetArgsRetsAndTypesFromFunctionGraph() local