Home
last modified time | relevance | path

Searched refs:arg_node (Results 1 – 12 of 12) sorted by relevance

/external/llvm/test/CodeGen/X86/
Dcombiner-aa-1.ll8 %struct.arg_node = type { %struct.node, i8*, %struct.alst_node* }
9 %struct.arglst_node = type { %struct.alst_node, %struct.arg_node*, %struct.arglst_node* }
10 %struct.lam_node = type { %struct.alst_node, %struct.arg_node*, %struct.alst_node* }
13 …ERK8exp_node._ZN11arglst_nodeD0Ev(%struct.lam_node* %this.this, %struct.arg_node* %outer_arg, %str…
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc136 TF_ASSIGN_OR_RETURN(Node * arg_node, in BuildLoopCondition()
139 node_map[arg.enter->id()] = arg_node; in BuildLoopCondition()
141 node_map[arg.merge->id()] = arg_node; in BuildLoopCondition()
179 TF_ASSIGN_OR_RETURN(Node * arg_node, BuildArgNode(output, dtype, i)); in BuildLoopBody()
183 node_map[arg.enter->id()] = arg_node; in BuildLoopBody()
184 output->AddEdge(arg_node, 0, retval_node, 0); in BuildLoopBody()
194 node_map[arg.switch_node->id()] = arg_node; in BuildLoopBody()
Dgraph_compiler_util.cc80 Node* arg_node = nullptr; in AddArgNodes() local
90 .Finalize(graph, &arg_node)); in AddArgNodes()
91 arg_nodes->insert(arg_node); in AddArgNodes()
105 graph->AddEdge(arg_node, 0, edge->dst(), edge->dst_input()); in AddArgNodes()
Dtf2xla_util.cc132 Node* arg_node = arg_nodes[arg_index]; in ReplaceArgUsageWithConstNode() local
139 for (const Edge* e : arg_node->out_edges()) { in ReplaceArgUsageWithConstNode()
/external/mesa3d/src/compiler/nir/
Dnir_lower_vars_to_ssa.c518 struct deref_node *arg_node = get_deref_node(arg_deref, state); in lower_copies_to_load_store() local
521 if (arg_node == NULL || arg_node == node) in lower_copies_to_load_store()
524 struct set_entry *arg_entry = _mesa_set_search(arg_node->copies, copy); in lower_copies_to_load_store()
526 _mesa_set_remove(arg_node->copies, arg_entry); in lower_copies_to_load_store()
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtemplates_test.py301 arg_node = node[0].value.args[1].value
302 self.assertIsInstance(arg_node.generators[0].target.ctx, gast.Store)
303 self.assertIsInstance(arg_node.elt.ctx, gast.Load)
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc503 auto add_conversion = [this](Node* arg_node, bool stacked) { in AddArgTensorMappings() argument
506 arg_node->attrs().Find("index")->i(), &input_node)); in AddArgTensorMappings()
508 conversion_map_.insert({{arg_node, 0}, {input_node, 0, stacked}}); in AddArgTensorMappings()
511 conversion_map_.insert({{arg_node, Graph::kControlSlot}, in AddArgTensorMappings()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc671 Node* arg_node = xla_graph->AddNode(arg_def, &s); in MoveHeadOutsideCompilationToHost() local
677 xla_graph->AddEdge(arg_node, 0, dst, dst_input); in MoveHeadOutsideCompilationToHost()
1787 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1788 for (const Edge* e : arg_node->out_edges()) { in FindArgsToLiftForWhileNode()
1819 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1821 arg_node->out_edges().begin(), arg_node->out_edges().end(), in FindArgsToLiftForWhileNode()
1830 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1831 for (const Edge* e : arg_node->out_edges()) { in FindArgsToLiftForWhileNode()
1860 const Node* arg_node = fbody.arg_nodes[i]; in FindArgsToLiftForCallNode() local
1861 if (arg_node->out_edges().empty()) { in FindArgsToLiftForCallNode()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc155 const Node* arg_node = arg_index_node_pair.second; in Initialize() local
156 arg_output_locations_[arg_index].reserve(arg_node->out_edges().size()); in Initialize()
157 for (const Edge* e : arg_node->out_edges()) { in Initialize()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc523 Node* arg_node = function_body.graph->AddNode(arg_def, &s); in AddOutsideCompilationInputArgToFunctionBody() local
525 return arg_node; in AddOutsideCompilationInputArgToFunctionBody()
532 Node* arg_node) { in AddMatchingRetvalNode() argument
536 ret_builder.Input(arg_node->name(), 0, data_type); in AddMatchingRetvalNode()
542 function_body.graph->AddEdge(arg_node, 0, ret_node, 0); in AddMatchingRetvalNode()
550 Node* arg_node) { in ReplaceLiftedArgNodePlaceholderWithArg() argument
560 function_body.graph->AddControlEdge(arg_node, e->dst()); in ReplaceLiftedArgNodePlaceholderWithArg()
562 function_body.graph->AddEdge(arg_node, 0, e->dst(), e->dst_input()); in ReplaceLiftedArgNodePlaceholderWithArg()
639 TF_ASSIGN_OR_RETURN(Node * arg_node, in PostprocessLiftedArgsForWhile()
644 AddMatchingRetvalNode(*body_function_body, i, data_types[i], arg_node)); in PostprocessLiftedArgsForWhile()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1518 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local
1521 mlir::Operation* island = node_values_.find(arg_node.node->id())->second; in ConvertFunctionArgAndRets()
1531 arg_nodes_to_values.try_emplace({arg_node.node, arg_node.index}, arg_def); in ConvertFunctionArgAndRets()
1538 if (!arg_node.node->requested_device().empty()) in ConvertFunctionArgAndRets()
1540 arg_node.node->requested_device())); in ConvertFunctionArgAndRets()
1542 if (arg_node.node->IsArg()) { in ConvertFunctionArgAndRets()
1544 set_attributes_on_func(arg_node.node, i, /*is_arg=*/true)); in ConvertFunctionArgAndRets()
2374 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local
2375 if (arg_node == nullptr) { in InferMainFunctionType()
2385 imported_dtype = arg_node->output_type(0); in InferMainFunctionType()
[all …]
/external/python/jinja/src/jinja2/
Dparser.py870 arg_node = self.parse_primary()
871 arg_node = self.parse_postfix(arg_node)
872 args = [arg_node]