Searched defs:arg_node (Results 1 – 8 of 8) sorted by relevance
80 Node* arg_node = nullptr; in AddArgNodes() local
132 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
155 const Node* arg_node = arg_index_node_pair.second; in Initialize() local
503 auto add_conversion = [this](Node* arg_node, bool stacked) { in AddArgTensorMappings()
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()
1518 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local2374 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local2460 auto& arg_node = arg_node_and_idx.value(); in GetArgsRetsAndTypesFromFunctionGraph() local