| /external/mesa3d/src/compiler/nir/ |
| D | nir_opt_find_array_copies.c | 437 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/ |
| D | collective_order.cc | 101 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/ |
| D | export_graphdef.cc | 158 Status AddEdgeBetweenNodes(Value src, Node* dst_node, unsigned dst_index);
267 Status Exporter::AddEdgeBetweenNodes(Value src, Node* dst_node, in AddEdgeBetweenNodes() argument
282 graph_->AddControlEdge(node_it->second, dst_node); in AddEdgeBetweenNodes()
284 graph_->AddEdge(node_it->second, input_result.getResultNumber(), dst_node, in AddEdgeBetweenNodes()
295 graph_->AddEdge(input_node_it->second, 0, dst_node, dst_index); in AddEdgeBetweenNodes()
307 auto* dst_node = returns_[fetch][operand_and_idx.index()]; in AddEdge() local
308 TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(operand, dst_node, 0)); in AddEdge()
318 auto* dst_node = nodes_[inst]; in AddEdge() local
320 AddEdgeBetweenNodes(next_iter_sink.input(), dst_node, 0)); in AddEdge()
322 TF_RETURN_IF_ERROR(AddEdgeBetweenNodes(control_and_idx.value(), dst_node, in AddEdge()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | grappler_test.cc | 199 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/ |
| D | functionalize_cond.cc | 1233 Node* dst_node = oe->dst(); in RemoveRedundantMerge() local
1240 dst_node, dst_port); in RemoveRedundantMerge()
1286 Node* dst_node = e->dst(); in RemoveRedundantSwitch() local
1291 if (IsMerge(dst_node)) { in RemoveRedundantSwitch()
1292 auto id_or = JoinCondStatesMerge(dst_node, dst_id, in RemoveRedundantSwitch()
1293 state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1295 FormatNodeForError(*dst_node)); in RemoveRedundantSwitch()
1296 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
1299 JoinCondStatesNonMerge(dst_id, state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1301 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 374 utils::MutableNodeView* dst_node, in UpdateFaninEdgesWithOp() argument
376 const bool is_in_frame = context->frames.IsInFrame(*dst_node->node()); in UpdateFaninEdgesWithOp()
378 auto& fanin_port = dst_node->GetRegularFanin(dst_port); in UpdateFaninEdgesWithOp()
383 GetFaninNameFormat(dst_node->GetName(), dst_port, in UpdateFaninEdgesWithOp()
387 dst_port, fanin_node_view, dst_node)); in UpdateFaninEdgesWithOp()
487 utils::MutableNodeView* src_node, utils::MutableNodeView* dst_node) { in UpdateEdge() argument
489 DCHECK(dst_node != nullptr); in UpdateEdge()
491 auto* dst_node_def = dst_node->node(); in UpdateEdge()
499 ->GetInputProperties(dst_node->GetName())[dst_port] in UpdateEdge()
549 mutation->AddOrUpdateRegularFanin(dst_node, dst_port, {added_node_name, 0}); in UpdateEdge()
|
| D | generic_layout_optimizer_transposer.h | 139 utils::MutableNodeView* dst_node,
186 utils::MutableNodeView* dst_node);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 381 : src_name(name), src_index(i1), dst_node(n), dst_index(i2) {} in EdgeInfo()
385 Node* dst_node; member
1304 g_->AddControlEdge(src_node, e.dst_node, kDoNotCheckDuplicates); in AddBackEdges()
1307 MakeEdge(src_node, e.src_index, e.dst_node, e.dst_index)); in AddBackEdges()
1311 << e.dst_node->name(); in AddBackEdges()
|
| /external/vulkan-validation-layers/layers/ |
| D | descriptor_sets.cpp | 1717 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/ |
| D | encapsulate_subgraphs_pass.cc | 474 Node* dst_node = edge->dst(); in RecordArg() local
475 Node* dst_image = node_images.at(dst_node); in RecordArg()
477 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
|