Searched defs:arg_node (Results 1 – 13 of 13) sorted by relevance
80 Node* arg_node = nullptr; in AddArgNodes() local
188 Node* arg_node = arg_nodes[arg_index]; in ReplaceArgUsageWithConstNode() local
1745 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1777 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1788 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local1818 const Node* arg_node = fbody.arg_nodes[i]; in FindArgsToLiftForCallNode() local1904 Node* arg_node = fbody->arg_nodes[i]; in RemoveArgsToLiftFromFunctionBody() local
678 tensorflow::Node* arg_node = graph->AddNode(arg_node_def, &status); in PrepareGraphForMlir() local730 Node* arg_node = graph->AddNode(arg_node_def, &status); in PrepareGraphForMlir() local
1211 Node* arg_node = *it; in SelectGraphToExecute() local
169 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
58 Node* arg_node = nullptr; member
237 const Edge* FindHostToDeviceEdge(Node* arg_node) { in FindHostToDeviceEdge()573 Node* arg_node = input_edge->src(); in FindTpuReplicatedInputAndXlaSharding() local2071 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()
2740 struct cil_tree_node *arg_node = NULL; in cil_build_call_args() local
530 Node* arg_node) { in AddMatchingRetvalNode()546 Node* arg_node) { in ReplaceLiftedArgNodePlaceholderWithArg()
1642 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local2588 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local2697 auto& arg_node = arg_node_and_idx.value(); in GetArgsRetsAndTypesFromFunctionGraph() local