Home
last modified time | relevance | path

Searched refs:dst_node (Results 1 – 11 of 11) sorted by relevance

/external/mesa3d/src/compiler/nir/
Dnir_opt_find_array_copies.c437 struct match_node *dst_node = in handle_write() local
443 if (nir_src_as_uint((*instr)->arr.index) != dst_node->next_array_idx) in handle_write()
446 if (dst_node->next_array_idx == 0) { in handle_write()
451 nir_deref_path_init(&dst_node->first_src_path, src, state->dead_ctx); in handle_write()
455 bool result = try_match_deref(&dst_node->first_src_path, in handle_write()
456 &dst_node->src_wildcard_idx, in handle_write()
457 &src_path, dst_node->next_array_idx, in handle_write()
475 if (dst_node->last_successful_write < dst_node->last_overwritten) in handle_write()
478 dst_node->last_successful_write = write_index; in handle_write()
483 dst_node->next_array_idx++; in handle_write()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dcollective_order.cc101 Node* dst_node = collective_nodes[dst_idx]; in CreateControlDependencies() local
104 << dst_node->name() << " instance " << instance_keys[dst_idx]; in CreateControlDependencies()
105 (*dependency_edges)[src_node].insert(dst_node); in CreateControlDependencies()
107 src_paths.insert(dst_node); in CreateControlDependencies()
108 for (Node* downstream_node : all_paths[dst_node]) { in CreateControlDependencies()
154 for (Node* dst_node : pair.second) { in InsertControlDependencies()
155 graph->AddControlEdge(src_node, dst_node); in InsertControlDependencies()
166 for (Node* dst_node : pair.second) { in InsertControlDependencies()
167 wait_for[dst_node].insert(src_instance); in InsertControlDependencies()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc197 Status AddEdgeBetweenNodes(Value src, Node* dst_node, unsigned dst_index);
287 Status Exporter::AddEdgeBetweenNodes(Value src, Node* dst_node, in AddEdgeBetweenNodes() argument
302 graph_->AddControlEdge(node_it->second, dst_node); in AddEdgeBetweenNodes()
304 graph_->AddEdge(node_it->second, input_result.getResultNumber(), dst_node, in AddEdgeBetweenNodes()
315 graph_->AddEdge(input_node_it->second, 0, dst_node, dst_index); in AddEdgeBetweenNodes()
327 auto* dst_node = returns_[fetch][operand_and_idx.index()]; in AddEdge() local
328 TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(operand, dst_node, 0)); in AddEdge()
338 auto* dst_node = nodes_[inst]; in AddEdge() local
340 AddEdgeBetweenNodes(next_iter_sink.input(), dst_node, 0)); in AddEdge()
342 TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(control_and_idx.value(), dst_node, in AddEdge()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc199 const NodeDef* dst_node = node_map.GetNode(dst); in IsNodesDirectlyConnected() local
201 EXPECT_TRUE(dst_node != nullptr) << dst << " node not found"; in IsNodesDirectlyConnected()
202 return src_node && dst_node && dst_node->input(position) == src_node->name(); in IsNodesDirectlyConnected()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc1240 Node* dst_node = oe->dst(); in RemoveRedundantMerge() local
1247 dst_node, dst_port); in RemoveRedundantMerge()
1293 Node* dst_node = e->dst(); in RemoveRedundantSwitch() local
1298 if (IsMerge(dst_node)) { in RemoveRedundantSwitch()
1299 auto id_or = JoinCondStatesMerge(dst_node, dst_id, in RemoveRedundantSwitch()
1300 state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1302 FormatNodeForError(*dst_node)); in RemoveRedundantSwitch()
1303 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
1306 JoinCondStatesNonMerge(dst_id, state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1308 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc377 utils::MutableNodeView* dst_node, in UpdateFaninEdgesWithOp() argument
379 const bool is_in_frame = context->frames.IsInFrame(*dst_node->node()); in UpdateFaninEdgesWithOp()
381 auto& fanin_port = dst_node->GetRegularFanin(dst_port); in UpdateFaninEdgesWithOp()
386 GetFaninNameFormat(dst_node->GetName(), dst_port, in UpdateFaninEdgesWithOp()
390 dst_port, fanin_node_view, dst_node)); in UpdateFaninEdgesWithOp()
490 utils::MutableNodeView* src_node, utils::MutableNodeView* dst_node) { in UpdateEdge() argument
492 DCHECK(dst_node != nullptr); in UpdateEdge()
494 auto* dst_node_def = dst_node->node(); in UpdateEdge()
503 ->GetInputProperties(dst_node->GetName())[dst_port] in UpdateEdge()
554 mutation->AddOrUpdateRegularFanin(dst_node, dst_port, {added_node_name, 0}); in UpdateEdge()
Dgeneric_layout_optimizer_transposer.h132 utils::MutableNodeView* dst_node,
179 utils::MutableNodeView* dst_node);
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc798 const Node* dst_node = out_edge->dst(); in ClusterizeNodes() local
799 CHECK_NOTNULL(dst_node); in ClusterizeNodes()
800 const bool dst_is_outside = node_names.count(dst_node->name()) <= 0; in ClusterizeNodes()
804 if (dst_node->IsSink()) { in ClusterizeNodes()
984 Node* dst_node = edge->dst(); in FuseCluster() local
989 graph.AddEdge(fused_node, j, dst_node, dst_input); in FuseCluster()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc384 : src_name(name), src_index(i1), dst_node(n), dst_index(i2) {} in EdgeInfo()
388 Node* dst_node; member
1307 g_->AddControlEdge(src_node, e.dst_node, kDoNotCheckDuplicates); in AddBackEdges()
1310 MakeEdge(src_node, e.src_index, e.dst_node, e.dst_index)); in AddBackEdges()
1314 << e.dst_node->name(); in AddBackEdges()
/external/vulkan-validation-layers/layers/
Ddescriptor_sets.cpp1717 auto dst_node = GetSetNode(dst_set); in ValidateUpdateDescriptorSets() local
1720 assert(dst_node); in ValidateUpdateDescriptorSets()
1723 … if (!ValidateCopyUpdate(&p_cds[i], dst_node, src_node, func_name, &error_code, &error_str)) { in ValidateUpdateDescriptorSets()
1755 auto dst_node = dev_data->GetSetNode(dst_set); in PerformUpdateDescriptorSets() local
1756 if (src_node && dst_node) { in PerformUpdateDescriptorSets()
1757 dst_node->PerformCopyUpdate(&p_cds[i], src_node); in PerformUpdateDescriptorSets()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc495 Node* dst_node = edge->dst(); in RecordArg() local
496 Node* dst_image = node_images.at(dst_node); in RecordArg()
498 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()