Home
last modified time | relevance | path

Searched refs:src_output (Results 1 – 25 of 116) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Drearrange_function_argument.cc119 int src_output = e->src_output(); in ReorderInputEdges() local
123 g->AddEdge(src, src_output, n, new_dst_input)->DebugString(); in ReorderInputEdges()
143 int src_output = e->src_output(); in ReorderOutputEdges() local
144 int new_src_output = index_mapping.at(src_output); in ReorderOutputEdges()
154 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in ReorderOutputEdges()
240 int src_output = e->src_output(); in RearrangeOutputEdges() local
241 auto iter = retval_index_mapping.find(src_output); in RearrangeOutputEdges()
243 TF_RET_CHECK(resource_retval_to_arg.find(src_output) != in RearrangeOutputEdges()
248 n->input_edge(resource_retval_to_arg.at(src_output), &input_edge)); in RearrangeOutputEdges()
249 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RearrangeOutputEdges()
[all …]
H A Dfunctionalize_cond.cc130 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()
137 *val = OutputTensor(val_edge->src(), val_edge->src_output()); in GetSwitchValue()
200 explicit CondArgNode(Node* src, int src_output) in CondArgNode()
201 : src(src), src_output(src_output) {} in CondArgNode()
204 return absl::StrCat("src=", src->name(), ":", src_output, in ToString()
209 int src_output; member
429 std::pair<Node*, int> key = std::make_pair(e->src(), e->src_output()); in BuildArgumentNodes()
440 DataType dtype = cond_arg_node.src->output_type(cond_arg_node.src_output); in BuildArgumentNodes()
454 int branch_index = e->src_output(); in BuildArgumentNodes()
510 int src_output = edge->src_output(); in AddSwitchNodeAlongEdge() local
[all …]
H A Dfunctionalize_while.cc80 int a_src_output = a->src_output(), in CopySubgraph()
81 b_src_output = b->src_output(); in CopySubgraph()
99 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() in CopySubgraph() local
101 : e->src_output(); in CopySubgraph()
103 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input()); in CopySubgraph()
250 graph->AddEdge(e->src(), e->src_output(), new_arg.enter, in FunctionalizeLoop()
366 if (edge->src_output() == 0) { in FunctionalizeLoop()
454 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
469 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
H A Dtf2xla_util.cc129 Status ReplaceSrcOutputUsageWithNode(Graph* g, Node* src, int src_output, in ReplaceSrcOutputUsageWithNode() argument
131 VLOG(1) << "Replace usages of output " << src_output << " of node " in ReplaceSrcOutputUsageWithNode()
141 if (e->IsControlEdge() || e->src_output() != src_output) { in ReplaceSrcOutputUsageWithNode()
361 IsLoopInvariant(fbody, e->src_output(), lookup_fld, in TraverseUnmodifiedPathBackward()
364 VLOG(2) << "Non-loop-invariant: index " << e->src_output() << " of " in TraverseUnmodifiedPathBackward()
370 TF_RETURN_IF_ERROR(e->src()->input_edge(e->src_output(), &e)); in TraverseUnmodifiedPathBackward()
816 graph->AddEdge(edge->src(), edge->src_output(), new_node, in RewriteAssociatedFunction()
820 graph->AddEdge(new_node, edge->src_output(), edge->dst(), in RewriteAssociatedFunction()
896 {edge->dst(), edge->src_output(), edge->dst_input()}); in ReplaceNode()
904 g->AddEdge(in_edge->src(), in_edge->src_output(), new_node, in ReplaceNode()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dplacer_inspection_required_ops_utils.cc174 NodeDefBuilder::NodeOut input(edge->src()->name(), edge->src_output(), in AddInputIdentity()
182 << edge->src_output() << " -> " << edge->dst()->name() << ":" in AddInputIdentity()
187 graph->AddEdge(edge->src(), edge->src_output(), identity_node, 0); in AddInputIdentity()
205 auto add_identity = [&](int src_output, const string& identity_name, in AddOutputIdentities()
208 builder.Attr("T", node->output_type(src_output)); in AddOutputIdentities()
209 NodeDefBuilder::NodeOut input(node->name(), src_output, in AddOutputIdentities()
210 node->output_type(src_output)); in AddOutputIdentities()
217 graph->AddEdge(node, src_output, *identity_node, 0); in AddOutputIdentities()
233 output_used[edge->src_output()] = true; in AddOutputIdentities()
237 int src_output = edge->src_output(); in AddOutputIdentities() local
[all …]
H A Dmkl_tfconversion_pass.cc152 DataType src_datatype = src->output_type(e->src_output()); in InsertConversionNodeOnEdge()
159 std::to_string(e->src_output()) + " and " + dst->name() + in InsertConversionNodeOnEdge()
168 .Input(src, e->src_output()) in InsertConversionNodeOnEdge()
170 e->src_output(), in InsertConversionNodeOnEdge()
224 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
225 BaseType(edges[1]->src()->output_type(edges[1]->src_output()))); in InsertInputConversionNode()
226 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
239 .Input(edges[0]->src(), edges[0]->src_output()) in InsertInputConversionNode()
240 .Input(edges[1]->src(), edges[1]->src_output()) in InsertInputConversionNode()
241 .Input(edges[2]->src(), edges[2]->src_output()) in InsertInputConversionNode()
[all …]
H A Dmemory_types.cc83 MemoryType sm = gtl::FindWithDefault(out, {e->src()->id(), e->src_output()}, in ProcessMemoryTypes()
87 VLOG(1) << e->src()->id() << ":" << e->src_output() << " -> " in ProcessMemoryTypes()
103 e->src_output(), " and ", e->dst()->id(), ":", in ValidateMemoryTypes()
127 .Input(edge->src(), edge->src_output()) in Send()
144 .Attr("tensor_type", edge->src()->output_type(edge->src_output())) in Recv()
187 const bool has_ref = IsRefType(e->src()->output_type(e->src_output())); in EnsureMemoryTypes()
189 Endpoint key{e->src()->id(), e->src_output()}; in EnsureMemoryTypes()
H A Dshape_refiner.cc239 ic->SetInput(e->dst_input(), input_ic->output(e->src_output())); in AddNodeInternal()
242 input_ic->output_handle_shapes_and_types(e->src_output()); in AddNodeInternal()
244 DataType input_type = e->src()->output_type(e->src_output()); in AddNodeInternal()
324 int src_output = e->src_output(); in UpdateNode() local
338 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()
345 if (node_context->RelaxInput(dst_input, c->output(src_output))) { in UpdateNode()
360 if (e->src()->output_type(src_output) == DT_RESOURCE) { in UpdateNode()
361 auto* outputs = c->output_handle_shapes_and_types(src_output); in UpdateNode()
416 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
457 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
[all …]
H A Dfunction_utils.cc109 if (IsRefType(e->src()->output_type(e->src_output()))) { in GetTheOnlyDataEdge()
152 g->AddEdge(in->src(), in->src_output(), out->dst(), out->dst_input()); in RemoveIdentityNodes()
211 *id_node = identity("input", {e->src(), e->src_output()}); in RemoveListArrayConverter()
235 Node* id_node = identity_nodes[e->src_output()]; in RemoveListArrayConverter()
238 << e->src_output(); in RemoveListArrayConverter()
353 } else if (e->src_output() == 0) { in ToGraphDef()
356 ndef->add_input(strings::StrCat(srcname, ":", e->src_output())); in ToGraphDef()
H A Dconstant_folding.cc77 CHECK_GT(known_shape.size(), in->src_output()) << known_shape_iter->first; in ReadPartialShapesFromShapeMap()
80 (*input_shapes)[in->dst_input()] = known_shape[in->src_output()]; in ReadPartialShapesFromShapeMap()
395 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], in AddNodeToConstantGraph()
402 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], in AddNodeToConstantGraph()
467 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
468 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
475 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
476 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
541 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
H A Dreplicate_per_replica_nodes.cc73 graph->AddEdge(src, edge->src_output(), dst, edge->dst_input()); in ReplicateFromRegularDeviceToCompositeDevice()
101 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), dst, in ReplicateFromCompositeDeviceToCompositeDevice()
123 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), dst, in ReplicateFromCompositeDeviceToRegularDevice()
158 const DataType dtype = edge->src()->output_type(edge->src_output()); in ReplicateFromCompositeDeviceToRegularDevice()
164 replicated_node->name(), edge->src_output(), dtype}); in ReplicateFromCompositeDeviceToRegularDevice()
172 graph->AddEdge(src_replicated_nodes[i], edge->src_output(), pack_node, in ReplicateFromCompositeDeviceToRegularDevice()
H A Dlower_case_op.cc64 Status AddInput(Node* src, int src_output);
146 Status CaseBuilder::AddInput(Node* src, int src_output) { in AddInput() argument
160 .Input(src, src_output) in AddInput()
179 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output())); in AddInputs()
247 graph_->AddEdge(merges[e->src_output()], 0, e->dst(), e->dst_input()); in AddOutputs()
H A Dlower_if_op.cc66 Status AddInput(Node* src, int src_output);
179 Status CondBuilder::AddInput(Node* src, int src_output) { in AddInput() argument
197 .Input(src, src_output) in AddInput()
215 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output())); in AddInputs()
275 graph_->AddEdge(merges[e->src_output()], 0, e->dst(), e->dst_input()); in AddOutputs()
H A Dsingle_threaded_executor.cc172 if (e->src_output() == Graph::kControlSlot) { in Initialize()
174 } else if (e->src_output() != 0) { in Initialize()
175 return errors::Internal("Invalid output index ", e->src_output(), in Initialize()
191 if (e->src_output() == Graph::kControlSlot) { in Initialize()
193 } else if (e->src_output() != 0) { in Initialize()
194 return errors::Internal("Invalid output index ", e->src_output(), in Initialize()
215 kernel_state.output_locations[e->src_output()].push_back( in Initialize()
H A Dgraph_view.cc165 CHECK_LE(e->src_output(), 0x3FFFFFFF); // Must fit in 31 bits in InitializeNode()
166 dst_edge->output_slot = e->src_output(); in InitializeNode()
339 e->src_output(), &attr)) { in SetScopedAllocatorAttrs()
341 (use_attrs + e->src_output())->Merge(attr); in SetScopedAllocatorAttrs()
343 attr = *(use_attrs + e->src_output()); in SetScopedAllocatorAttrs()
441 attrs[e->src_output()].Merge(attr); in SetAllocAttrs()
H A Dgradients.cc115 n_inputs[e->dst_input()] = {e->src(), e->src_output()}; in AddSymGrad()
296 BackpropAlongEdge(y_grad_node_outputs_[i], {e->src(), e->src_output()}); in InitBackprop()
376 BackpropZerosAlongEdge({e->src(), e->src_output()}); in Compute()
387 graph_->AddEdge(e->src(), e->src_output(), grad, e->dst_input()); in Compute()
396 BackpropAlongEdge({grad, e->dst_input()}, {e->src(), e->src_output()}); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc295 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RemoveIdentityNodesForArgRetval()
463 g->AddEdge(original_edge->src(), original_edge->src_output(), in MoveHeadOutsideCompilationToHost()
466 g->AddEdge(node_images[e->src()][replica_id], e->src_output(), in MoveHeadOutsideCompilationToHost()
498 new_arg_types.push_back(e->src()->output_type(e->src_output())); in MoveHeadOutsideCompilationToHost()
572 int src_output = e->src_output(); in MoveHeadOutsideCompilationToHost() local
575 g->AddEdge(src, src_output, xla_node, input_index); in MoveHeadOutsideCompilationToHost()
584 per_replica_inputs.push_back(std::make_pair(e->src(), e->src_output())); in MoveHeadOutsideCompilationToHost()
596 int src_output = in MoveHeadOutsideCompilationToHost() local
601 g->AddEdge(src, src_output, xla_node, in MoveHeadOutsideCompilationToHost()
611 ->src_output(); in MoveHeadOutsideCompilationToHost()
[all …]
H A Dtpu_embedding_rewrite_pass_utils.cc37 edge->src()->name(), edge->src_output(), in ReplaceNode()
44 edge->src()->name(), edge->src_output(), in ReplaceNode()
59 new_edge_src->name(), edge->src_output(), in ReplaceNode()
61 graph->AddEdge(new_edge_src, edge->src_output(), new_edge_dst, in ReplaceNode()
H A Ddistributed_tpu_rewrite_pass.cc572 int src_output = edge->src_output(); in ReplaceCompilationResultNodeWithIdentity() local
575 if (src_output == Graph::kControlSlot) { in ReplaceCompilationResultNodeWithIdentity()
578 graph->AddEdge(id_node, src_output, dst, dst_input); in ReplaceCompilationResultNodeWithIdentity()
1104 /*input=*/{resource->src(), resource->src_output()}, in CreateOrGetXlaSplitNodeForVariableArg()
1900 TF_RET_CHECK(it->second.size() > edge.src_output()); in GetEdgeShape()
1901 *info = &it->second[edge.src_output()]; in GetEdgeShape()
2156 partitioned_output_nodes[edge->src_output()] = out_edge->dst(); in AssignArgsAndRetvalsToCores()
2488 int src_output; in BuildDynamicShapeNodes() local
2509 src_output = variable_input->src_output(); in BuildDynamicShapeNodes()
2520 src_output = replicate_input_edges[i]->src_output(); in BuildDynamicShapeNodes()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dincrease_dynamism_for_auto_jit_pass.cc105 Output(slice_input_edge->src(), slice_input_edge->src_output()); in GetSliceInputs()
107 Output(slice_begin_edge->src(), slice_begin_edge->src_output()); in GetSliceInputs()
109 Output(slice_size_edge->src(), slice_size_edge->src_output()); in GetSliceInputs()
287 DCHECK(e->src_output() == 0 || e->src_output() == Graph::kControlSlot); in ReplaceTensorFlowSliceWithStaticShapedSlice()
289 int src_output = e->src_output(); in ReplaceTensorFlowSliceWithStaticShapedSlice() local
293 g->AddEdge(static_shaped_slice, src_output, dst, dst_input); in ReplaceTensorFlowSliceWithStaticShapedSlice()
H A Dencapsulate_xla_computations_pass.cc214 e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":", in Encapsulate()
272 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
280 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
293 TF_RET_CHECK(le->src_output() < num_outputs); in BuildXlaLaunchOps()
303 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()
309 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()
312 output_types[le->src_output()] = output_node->input_type(0); in BuildXlaLaunchOps()
H A Dencapsulate_util.cc147 int src_output = e->src_output(), dst_input = e->dst_input(); in PreprocessDataEdgesBetweenOutsideCompilations() local
152 absl::StrCat(src->name(), "_oc_to_oc_placeholder_", src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
153 auto placeholder_index = std::make_pair(src->name(), src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
158 placeholder_builder.Attr("dtype", src->output_type(src_output)); in PreprocessDataEdgesBetweenOutsideCompilations()
168 src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
232 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
H A Dbuild_xla_ops_pass.cc72 g->AddEdge(new_node, edge->src_output(), edge->dst(), edge->dst_input()); in MoveOutgoingEdges()
130 int oidx = e->src_output(); in MergeOutgoingDataEdges()
225 return Output(e->src(), e->src_output()); in IncomingEdgeAsOutput()
308 args[e->dst_input()] = Output(e->src(), e->src_output()); in ReplaceFunctionCallWithPartitionedCall()
333 g->AddEdge(call.operation.node(), e->src_output(), e->dst(), in ReplaceFunctionCallWithPartitionedCall()
434 if (e->src()->output_type(e->src_output()) == DT_INT32) { in PredicateInt32Inputs()
437 if (source_output_mem_types[e->src_output()] == DEVICE_MEMORY) { in PredicateInt32Inputs()
438 int32_inputs.push_back(Output(e->src(), e->src_output())); in PredicateInt32Inputs()
H A Dencapsulate_subgraphs_pass.cc423 return results_.at(OutputTensor(edge->src(), edge->src_output())); in GetResultIndexForEdge()
453 int src_slot = edge->src_output(); in RecordArg()
496 int src_slot = edge->src_output(); in RecordResult()
699 g->AddEdge(src_image, edge->src_output(), dst_image, edge->dst_input()); in CopySubgraphEdges()
707 DataType dtype = edge->src()->output_type(edge->src_output()); in CopySubgraphEdges()
712 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
735 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
850 return edge->src_output(); in FindOutputSlotOfEdgeSrc()
912 int src_output = FindOutputSlotOfEdgeSrc(src_func_id, dst_func_id, edge); in CopyEdgeToOutputGraph() local
918 ->emplace(OutputTensor(src_image, src_output), in CopyEdgeToOutputGraph()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_graph_utils.cc137 const int src_output = edge->src_output(); in InsertNodes() local
138 if (output_slot_to_edges.find(src_output) == output_slot_to_edges.end()) { in InsertNodes()
139 output_slot_to_edges[src_output] = {edge}; in InsertNodes()
141 output_slot_to_edges[src_output].push_back(edge); in InsertNodes()
301 const string& src_node_name, const int src_output, const DataType src_dt, in CreateCopyNode() argument
310 const string copy_node_name = GetCopyNodeName(src_node_name, src_output); in CreateCopyNode()
334 .Input(src_node_name, src_output, src_dt) in CreateCopyNode()

12345