Home
last modified time | relevance | path

Searched refs:arg_node (Results 1 – 10 of 10) 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.cc78 Node* arg_node = nullptr; in AddArgNodes() local
88 .Finalize(graph, &arg_node)); in AddArgNodes()
102 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/kernels/data/
Dsingle_threaded_executor.cc132 const Node* arg_node = arg_index_node_pair.second; in Initialize() local
133 arg_output_locations_[arg_index].reserve(arg_node->out_edges().size()); in Initialize()
134 for (const Edge* e : arg_node->out_edges()) { in Initialize()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc521 Node* arg_node = function_body.graph->AddNode(arg_def, &s); in AddOutsideCompilationInputArgToFunctionBody() local
523 return arg_node; in AddOutsideCompilationInputArgToFunctionBody()
530 Node* arg_node) { in AddMatchingRetvalNode() argument
534 ret_builder.Input(arg_node->name(), 0, data_type); in AddMatchingRetvalNode()
540 function_body.graph->AddEdge(arg_node, 0, ret_node, 0); in AddMatchingRetvalNode()
548 Node* arg_node) { in ReplaceLiftedArgNodePlaceholderWithArg() argument
558 function_body.graph->AddControlEdge(arg_node, e->dst()); in ReplaceLiftedArgNodePlaceholderWithArg()
560 function_body.graph->AddEdge(arg_node, 0, e->dst(), e->dst_input()); in ReplaceLiftedArgNodePlaceholderWithArg()
623 TF_ASSIGN_OR_RETURN(Node * arg_node, in PostprocessLiftedArgsForWhile()
628 AddMatchingRetvalNode(*body_function_body, i, data_types[i], arg_node)); in PostprocessLiftedArgsForWhile()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1206 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local
1209 mlir::Operation* island = node_values_.find(arg_node.node->id())->second; in ConvertFunctionArgAndRets()
1219 arg_nodes_to_values.try_emplace({arg_node.node, arg_node.index}, arg_def); in ConvertFunctionArgAndRets()
1226 if (!arg_node.node->requested_device().empty()) in ConvertFunctionArgAndRets()
1229 builder_.getStringAttr(arg_node.node->requested_device())); in ConvertFunctionArgAndRets()
1959 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local
1960 if (arg_node == nullptr) { in InferMainFunctionType()
1970 imported_dtype = arg_node->output_type(0); in InferMainFunctionType()
2042 auto& arg_node = arg_node_and_idx.value(); in GetArgsRetsAndTypesFromFunctionGraph() local
2043 if (arg_node.node == nullptr) in GetArgsRetsAndTypesFromFunctionGraph()
[all …]