Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/
Dactivity_test.py263 if_node = node.body[0]
265 anno.getanno(if_node, NodeAnno.BODY_SCOPE), ('x', 'y'), ('x', 'y', 'z'))
267 anno.getanno(if_node, NodeAnno.BODY_SCOPE).parent, ('x', 'y', 'z', 'u'),
270 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE), ('x', 'y'),
273 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE).parent,
288 if_node = node.body[0]
290 anno.getanno(if_node, NodeAnno.BODY_SCOPE), ('a', 'a.c'), ('a.b', 'd'))
292 anno.getanno(if_node, NodeAnno.ORELSE_SCOPE), ('a', 'a.c'),
295 anno.getanno(if_node, NodeAnno.BODY_SCOPE).parent, ('a', 'a.c', 'd'),
310 if_node = node.body[0]
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_opt_if.c105 nir_cf_node *if_node = nir_cf_node_next(&header_block->cf_node); in opt_peel_loop_initial_if() local
106 if (!if_node || if_node->type != nir_cf_node_if) in opt_peel_loop_initial_if()
109 nir_if *nif = nir_cf_node_as_if(if_node); in opt_peel_loop_initial_if()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_while_test.cc254 Node* if_node; in TEST() local
262 .Finalize(root.graph(), &if_node)); in TEST()
263 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
301 TF_ASSERT_OK(session.Run(feeds, {Output(if_node)}, &out_tensors)); in TEST()
310 TF_ASSERT_OK(session.Run(feeds, {Output(if_node)}, &out_tensors)); in TEST()
Dlower_if_op_test.cc187 Node* if_node; in TEST() local
197 .Finalize(root.graph(), &if_node)); in TEST()
200 root.WithOpName("read").WithControlDependencies(Output(if_node)), var, in TEST()
203 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_if_conversion.cpp120 if_node *nif = static_cast<if_node*>(nd1->first); in check_and_convert()
210 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()
Dsb_gcm.cpp271 if_node *f = static_cast<if_node*>(n); in sched_late()
Dsb_shader.h369 if_node* create_if();
Dsb_dump.cpp277 bool dump::visit(if_node& n, bool enter) { in visit()
Dsb_bc_finalize.cpp189 if_node *n_if = static_cast<if_node*>(repdep1->first); in finalize_if()
Dsb_ssa_builder.cpp273 bool ssa_rename::visit(if_node& n, bool enter) { in visit()
Dsb_ir.h1092 class if_node : public container_node {
1094 if_node() : container_node(NT_IF, NST_LIST), cond() {}; in if_node() function
Dsb_ir.cpp48 bool if_node::accept(vpass& p, bool enter) { return p.visit(*this, enter); } in accept()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc408 auto* if_node = fused_function->add_node_def(); in LazyConjunctionNodes() local
410 TF_CHECK_OK(if_builder.Finalize(if_node)); in LazyConjunctionNodes()
411 function_utils::SetUniqueFunctionNodeName("cond", fused_function, if_node); in LazyConjunctionNodes()
413 GetMutableOutputNode(fused_function, 0) = if_node->name() + ":output:0"; in LazyConjunctionNodes()
/external/tensorflow/tensorflow/python/autograph/pyct/
Dtransformer_test.py259 if_node = gast.If(gast.Name('x', gast.Load(), None), [node], [])
260 return if_node, if_node.body
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc222 Status PropagateConstIntoIfNode(Graph* g, Node* if_node, in PropagateConstIntoIfNode() argument
228 for (int i = 1; i < if_node->num_inputs(); i++) { in PropagateConstIntoIfNode()
230 TF_RETURN_IF_ERROR(if_node->input_node(i, &input_node)); in PropagateConstIntoIfNode()
244 if_node, attr_name, const_input_index_to_node, lookup_fld, fld)); in PropagateConstIntoIfNode()
Dfunctionalize_cond.h183 Status AddIdentityNode(const Node* replacee, Node* if_node, int port);
Dfunctionalize_cond.cc919 Status FunctionalizeCond::AddIdentityNode(const Node* replacee, Node* if_node, in AddIdentityNode() argument
923 .Input(if_node, port) in AddIdentityNode()
925 state_map_.ResetCondId(id, state_map_.LookupCondId(if_node)); in AddIdentityNode()
926 state_map_.ResetAncestorId(id, state_map_.LookupAncestorId(if_node)); in AddIdentityNode()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc683 Node *if_node = node_name_index["if"]; in TEST_F() local
684 EXPECT_NE(if_node, nullptr); in TEST_F()
687 GetNodeAttr(if_node->def(), "_xla_token_input_nodes", &token_inputs)); in TEST_F()
Dextract_outside_compilation_pass.cc939 Node* if_node = host_graph.AddNode(if_def, &s); in BuildHostGraphForIfNode() local
941 host_graph.AddEdge(recv_pred_node, 0, if_node, 0); in BuildHostGraphForIfNode()
942 host_graph.AddEdge(key_placeholder, 0, if_node, 1); in BuildHostGraphForIfNode()

12