Home
last modified time | relevance | path

Searched refs:if_node (Results 1 – 25 of 35) sorted by relevance

12

/external/mesa3d/src/compiler/nir/
Dnir_lower_memory_model.c193 foreach_list_typed(nir_cf_node, if_node, node, &nif->then_list) in lower_make_visible()
194 progress |= lower_make_visible(if_node, &cur_modes_then); in lower_make_visible()
195 foreach_list_typed(nir_cf_node, if_node, node, &nif->else_list) in lower_make_visible()
196 progress |= lower_make_visible(if_node, &cur_modes_else); in lower_make_visible()
232 foreach_list_typed_reverse(nir_cf_node, if_node, node, &nif->then_list) in lower_make_available()
233 progress |= lower_make_available(if_node, &cur_modes_then); in lower_make_available()
234 foreach_list_typed_reverse(nir_cf_node, if_node, node, &nif->else_list) in lower_make_available()
235 progress |= lower_make_available(if_node, &cur_modes_else); in lower_make_available()
Dnir_opt_if.c155 nir_cf_node *if_node = nir_cf_node_next(&header_block->cf_node); in opt_peel_loop_initial_if() local
156 if (!if_node || if_node->type != nir_cf_node_if) in opt_peel_loop_initial_if()
159 nir_if *nif = nir_cf_node_as_if(if_node); in opt_peel_loop_initial_if()
814 nir_cf_node *if_node = nir_cf_node_prev(&last_block->cf_node); in opt_if_loop_last_continue() local
815 while (if_node) { in opt_if_loop_last_continue()
816 if (if_node->type == nir_cf_node_if) { in opt_if_loop_last_continue()
817 nif = nir_cf_node_as_if(if_node); in opt_if_loop_last_continue()
838 if_node = nir_cf_node_prev(if_node); in opt_if_loop_last_continue()
852 nir_cf_extract(&tmp, nir_after_cf_node(if_node), in opt_if_loop_last_continue()
Dnir_to_lcssa.c151 nir_if *if_node = nir_cf_node_as_if(prev); in phi_is_invariant() local
152 if (!def_is_invariant(if_node->condition.ssa, loop)) in phi_is_invariant()
/external/llvm-project/polly/lib/External/isl/
Disl_ast_graft.c304 isl_ast_node *if_node; in ast_node_insert_if() local
309 if_node = isl_ast_node_alloc_if(expr); in ast_node_insert_if()
310 return isl_ast_node_if_set_then(if_node, node); in ast_node_insert_if()
542 static int clear_if_nodes(struct isl_if_node *if_node, int first, int n) in clear_if_nodes() argument
547 isl_set_free(if_node[i].guard); in clear_if_nodes()
548 isl_set_free(if_node[i].complement); in clear_if_nodes()
579 struct isl_if_node *if_node = NULL; in insert_pending_guard_nodes() local
591 if_node = isl_alloc_array(ctx, struct isl_if_node, n - 1); in insert_pending_guard_nodes()
592 if (!if_node) in insert_pending_guard_nodes()
616 if_node[j].guard); in insert_pending_guard_nodes()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops_test.cc266 Node* if_node; in TEST() local
274 .Finalize(root.graph(), &if_node)); in TEST()
275 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
313 TF_ASSERT_OK(session.Run(feeds, {Output(if_node)}, &out_tensors)); in TEST()
322 TF_ASSERT_OK(session.Run(feeds, {Output(if_node)}, &out_tensors)); in TEST()
Dlower_if_op_test.cc196 Node* if_node; in TEST() local
207 .Finalize(root.graph(), &if_node)); in TEST()
210 root.WithOpName("read").WithControlDependencies(Output(if_node)), var, in TEST()
213 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
/external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/
Dactivity_test.py307 if_node = node.body[0]
309 anno.getanno(if_node, NodeAnno.BODY_SCOPE), ('x', 'y'), ('x', 'y', 'z'))
311 anno.getanno(if_node, NodeAnno.BODY_SCOPE).parent, ('x', 'y', 'z', 'u'),
314 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE), ('x', 'y'),
317 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE).parent,
332 if_node = node.body[0]
334 anno.getanno(if_node, NodeAnno.BODY_SCOPE), ('a', 'a.c'), ('a.b', 'd'))
336 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE), ('a', 'a.c'),
339 anno.getanno(if_node, NodeAnno.BODY_SCOPE).parent, ('a', 'a.c', 'd'),
354 if_node = node.body[0]
[all …]
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_if_conversion.cpp117 if_node *nif = static_cast<if_node*>(nd1->first); in check_and_convert()
207 if_node *nif = static_cast<if_node*>(nd1->first); in run_on()
Dsb_def_use.cpp151 if_node *i = static_cast<if_node*>(n); in process_uses()
Dsb_gvn.cpp212 if_node &i = (if_node&)n; in process_op()
Dsb_liveness.cpp170 bool liveness::visit(if_node& n, bool enter) { in visit()
308 if_node &in = (if_node&)n; in process_ins()
Dsb_pass.h69 virtual bool visit(if_node &n, bool enter);
187 virtual bool visit(if_node &n, bool enter);
383 virtual bool visit(if_node &n, bool enter);
662 virtual bool visit(if_node &n, bool enter);
Dsb_shader.cpp328 if_node* shader::create_if() { in create_if()
329 if_node* n = new (pool.allocate(sizeof(if_node))) if_node(); in create_if()
Dsb_pass.cpp82 bool vpass::visit(if_node& n, bool enter) { return true; } in visit()
/external/tensorflow/tensorflow/compiler/jit/
Drearrange_function_argument_pass_test.cc137 const Node *if_node = node_name_index.at("if"); in TEST() local
138 ASSERT_NE(if_node, nullptr); in TEST()
140 TF_CHECK_OK(if_node->input_node(1, &input_node)); in TEST()
142 TF_CHECK_OK(if_node->input_node(2, &input_node)); in TEST()
Dcompilability_check_util.cc231 const Node& if_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableIf() argument
238 if_node, "then_branch", "if_then", encapsulating_function, lib_runtime, in IsCompilableIf()
243 if_node, "else_branch", "if_else", encapsulating_function, lib_runtime, in IsCompilableIf()
Dcompilability_check_util.h213 bool IsCompilableIf(const Node& if_node, FunctionLibraryRuntime* lib_runtime,
Dcompilability_check_util_test.cc297 Node* if_node; in TEST_F() local
305 .Finalize(root.graph(), &if_node)); in TEST_F()
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtransformer_test.py160 if_node = gast.If(
164 return if_node, if_node.body
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc409 auto* if_node = fused_function->add_node_def(); in LazyConjunctionNodes() local
411 TF_CHECK_OK(if_builder.Finalize(if_node)); in LazyConjunctionNodes()
412 function_utils::SetUniqueFunctionNodeName("cond", fused_function, if_node); in LazyConjunctionNodes()
414 GetMutableOutputNode(fused_function, 0) = if_node->name() + ":output:0"; in LazyConjunctionNodes()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc221 Status PropagateConstIntoIfNode(Graph* g, Node* if_node, in PropagateConstIntoIfNode() argument
227 for (int i = 1; i < if_node->num_inputs(); i++) { in PropagateConstIntoIfNode()
229 TF_RETURN_IF_ERROR(if_node->input_node(i, &input_node)); in PropagateConstIntoIfNode()
243 if_node, attr_name, const_input_index_to_node, lookup_fld, fld)); in PropagateConstIntoIfNode()
Dfunctionalize_cond.h192 Status AddIdentityNode(const Node* replacee, Node* if_node, int port);
Dfunctionalize_cond.cc960 Status FunctionalizeCond::AddIdentityNode(const Node* replacee, Node* if_node, in AddIdentityNode() argument
963 id_builder.Input(if_node, port); in AddIdentityNode()
965 if (GetNodeAttr(if_node->def(), kXlaOutsideCompilationAttrName, in AddIdentityNode()
972 state_map_.ResetCondId(id, state_map_.LookupCondId(if_node)); in AddIdentityNode()
973 state_map_.ResetAncestorId(id, state_map_.LookupAncestorId(if_node)); in AddIdentityNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1701 const Node& if_node, FunctionLibraryDefinition* fld) { in FindArgsToLiftForIfNode() argument
1704 TF_RETURN_IF_ERROR(GetNodeAttr(if_node.def(), "Tin", &dtypes)); in FindArgsToLiftForIfNode()
1717 InstantiateAssociatedFunction(if_node, "then_branch", fld)); in FindArgsToLiftForIfNode()
1721 InstantiateAssociatedFunction(if_node, "else_branch", fld)); in FindArgsToLiftForIfNode()
2195 Status LiftOutsideCompilationOnlyArgsFromIfNode(Graph* g, Node* if_node, in LiftOutsideCompilationOnlyArgsFromIfNode() argument
2201 FindArgsToLiftForIfNode(*if_node, fld)); in LiftOutsideCompilationOnlyArgsFromIfNode()
2205 TF_RETURN_IF_ERROR(GetNodeAttr(if_node->def(), "Tin", &dtypes)); in LiftOutsideCompilationOnlyArgsFromIfNode()
2220 args_to_lift, /*arg_to_input_edge_offset=*/1, g, if_node, in LiftOutsideCompilationOnlyArgsFromIfNode()
2225 InstantiateAssociatedFunction(*if_node, "then_branch", fld)); in LiftOutsideCompilationOnlyArgsFromIfNode()
2240 InstantiateAssociatedFunction(*if_node, "else_branch", fld)); in LiftOutsideCompilationOnlyArgsFromIfNode()
[all …]
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc259 for (Node* if_node : if_nodes) { in LowerFunctionalOps()
261 RewriteIfNode(if_node, g, /*keep_node_fetchable=*/false)); in LowerFunctionalOps()

12