Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/tf2xla/
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 …]
Dfunctionalize_cond.cc132 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()
139 *val = OutputTensor(val_edge->src(), val_edge->src_output()); in GetSwitchValue()
202 explicit CondArgNode(Node* src, int src_output) in CondArgNode()
203 : src(src), src_output(src_output) {} in CondArgNode()
206 return absl::StrCat("src=", src->name(), ":", src_output, in ToString()
211 int src_output; member
431 std::pair<Node*, int> key = std::make_pair(e->src(), e->src_output()); in BuildArgumentNodes()
442 DataType dtype = cond_arg_node.src->output_type(cond_arg_node.src_output); in BuildArgumentNodes()
456 int branch_index = e->src_output(); in BuildArgumentNodes()
512 int src_output = edge->src_output(); in AddSwitchNodeAlongEdge() local
[all …]
Dfunctionalize_while.cc82 int a_src_output = a->src_output(), in CopySubgraph()
83 b_src_output = b->src_output(); in CopySubgraph()
101 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() in CopySubgraph() local
103 : e->src_output(); in CopySubgraph()
105 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input()); in CopySubgraph()
251 graph->AddEdge(e->src(), e->src_output(), new_arg.enter, in FunctionalizeLoop()
367 if (edge->src_output() == 0) { in FunctionalizeLoop()
458 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
473 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
Dsharding_util.cc126 if (edge.src_output() < 0 || in ParseShardingFromEdgeSource()
127 edge.src_output() >= sharding.value().tuple_shardings_size()) { in ParseShardingFromEdgeSource()
133 sharding.value().tuple_shardings(edge.src_output()); in ParseShardingFromEdgeSource()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc178 NodeDefBuilder::NodeOut input(edge->src()->name(), edge->src_output(), in AddInputIdentity()
186 << edge->src_output() << " -> " << edge->dst()->name() << ":" in AddInputIdentity()
195 graph->AddEdge(edge->src(), edge->src_output(), identity_node, 0); in AddInputIdentity()
213 auto add_identity = [&](int src_output, const string& identity_name, in AddOutputIdentities()
216 builder.Attr("T", node->output_type(src_output)); in AddOutputIdentities()
217 NodeDefBuilder::NodeOut input(node->name(), src_output, in AddOutputIdentities()
218 node->output_type(src_output)); in AddOutputIdentities()
229 graph->AddEdge(node, src_output, *identity_node, 0); in AddOutputIdentities()
245 output_used[edge->src_output()] = true; in AddOutputIdentities()
249 int src_output = edge->src_output(); in AddOutputIdentities() local
[all …]
Dmkl_tfconversion_pass.cc153 DataType src_datatype = src->output_type(e->src_output()); in InsertConversionNodeOnEdge()
160 std::to_string(e->src_output()) + " and " + dst->name() + in InsertConversionNodeOnEdge()
169 .Input(src, e->src_output()) in InsertConversionNodeOnEdge()
171 e->src_output(), in InsertConversionNodeOnEdge()
225 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
226 BaseType(edges[1]->src()->output_type(edges[1]->src_output()))); in InsertInputConversionNode()
227 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
240 .Input(edges[0]->src(), edges[0]->src_output()) in InsertInputConversionNode()
241 .Input(edges[1]->src(), edges[1]->src_output()) in InsertInputConversionNode()
242 .Input(edges[2]->src(), edges[2]->src_output()) in InsertInputConversionNode()
[all …]
Dmemory_types.cc81 MemoryType sm = gtl::FindWithDefault(out, {e->src()->id(), e->src_output()}, in ProcessMemoryTypes()
85 VLOG(1) << e->src()->id() << ":" << e->src_output() << " -> " in ProcessMemoryTypes()
101 e->src_output(), " and ", e->dst()->id(), ":", in ValidateMemoryTypes()
125 .Input(edge->src(), edge->src_output()) in Send()
142 .Attr("tensor_type", edge->src()->output_type(edge->src_output())) in Recv()
185 const bool has_ref = IsRefType(e->src()->output_type(e->src_output())); in EnsureMemoryTypes()
187 Endpoint key{e->src()->id(), e->src_output()}; in EnsureMemoryTypes()
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()
Dreplicate_per_replica_nodes.cc63 graph->AddEdge(src, edge->src_output(), dst, edge->dst_input()); in ReplicateFromRegularDeviceToCompositeDevice()
84 graph->AddEdge(src_replicated_nodes.at(i), edge->src_output(), in ReplicateFromCompositeDeviceToCompositeDevice()
108 graph->AddEdge(replicated_node, edge->src_output(), dst, in ReplicateFromCompositeDeviceToRegularDevice()
129 const DataType dtype = edge->src()->output_type(edge->src_output()); in ReplicateFromCompositeDeviceToRegularDevice()
135 replicated_node->name(), edge->src_output(), dtype}); in ReplicateFromCompositeDeviceToRegularDevice()
145 graph->AddEdge(src_replicated_nodes[i], edge->src_output(), in ReplicateFromCompositeDeviceToRegularDevice()
Dshape_refiner.cc223 ic->SetInput(e->dst_input(), input_ic->output(e->src_output())); in AddNodeInternal()
226 input_ic->output_handle_shapes_and_types(e->src_output()); in AddNodeInternal()
228 DataType input_type = e->src()->output_type(e->src_output()); in AddNodeInternal()
308 int src_output = e->src_output(); in UpdateNode() local
322 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()
329 if (node_context->RelaxInput(dst_input, c->output(src_output))) { in UpdateNode()
344 if (e->src()->output_type(src_output) == DT_RESOURCE) { in UpdateNode()
345 auto* outputs = c->output_handle_shapes_and_types(src_output); in UpdateNode()
400 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
441 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
[all …]
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()
Dconstant_folding.cc76 CHECK_GT(known_shape.size(), in->src_output()) << known_shape_iter->first; in ReadPartialShapesFromShapeMap()
79 (*input_shapes)[in->dst_input()] = known_shape[in->src_output()]; in ReadPartialShapesFromShapeMap()
398 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], in AddNodeToConstantGraph()
405 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], in AddNodeToConstantGraph()
470 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
471 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
478 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
479 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
544 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
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()
Dgraph_view.cc163 CHECK_LE(e->src_output(), 0x3FFFFFFF); // Must fit in 31 bits in InitializeNode()
164 dst_edge->output_slot = e->src_output(); in InitializeNode()
322 e->src_output(), &attr)) { in SetScopedAllocatorAttrs()
324 (use_attrs + e->src_output())->Merge(attr); in SetScopedAllocatorAttrs()
326 attr = *(use_attrs + e->src_output()); in SetScopedAllocatorAttrs()
424 attrs[e->src_output()].Merge(attr); in SetAllocAttrs()
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()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc321 g->AddEdge(input_edge->src(), input_edge->src_output(), dst, dst_input); in RemoveIdentityNodesForArgRetval()
491 g->AddEdge(original_edge->src(), original_edge->src_output(), in MoveHeadOutsideCompilationToHost()
494 g->AddEdge(node_images[e->src()][replica_id], e->src_output(), in MoveHeadOutsideCompilationToHost()
526 new_arg_types.push_back(e->src()->output_type(e->src_output())); in MoveHeadOutsideCompilationToHost()
600 int src_output = e->src_output(); in MoveHeadOutsideCompilationToHost() local
603 g->AddEdge(src, src_output, xla_node, input_index); in MoveHeadOutsideCompilationToHost()
612 per_replica_inputs.push_back(std::make_pair(e->src(), e->src_output())); in MoveHeadOutsideCompilationToHost()
624 int src_output = in MoveHeadOutsideCompilationToHost() local
629 g->AddEdge(src, src_output, xla_node, in MoveHeadOutsideCompilationToHost()
639 ->src_output(); in MoveHeadOutsideCompilationToHost()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
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()
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()
234 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
Dshape_inference_helpers.cc36 BackEdge{e, e->src(), e->src_output(), e->dst(), e->dst_input()}); in Remove()
61 graph_->AddEdge(be.src, be.src_output, be.dst, be.dst_input); in Replace()
Dencapsulate_xla_computations_pass.cc222 e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":", in Encapsulate()
278 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
286 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
299 TF_RET_CHECK(le->src_output() < num_outputs); in BuildXlaLaunchOps()
308 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()
311 output_types[le->src_output()] = output_node->input_type(0); in BuildXlaLaunchOps()
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()
435 if (e->src()->output_type(e->src_output()) == DT_INT32) { in PredicateInt32Inputs()
438 if (source_output_mem_types[e->src_output()] == DEVICE_MEMORY) { in PredicateInt32Inputs()
439 int32_inputs.push_back(Output(e->src(), e->src_output())); in PredicateInt32Inputs()
Dclone_constants_for_better_clustering.cc49 g->AddEdge(e->src(), e->src_output(), new_in, e->dst_input()); in CloneNode()
125 TF_RET_CHECK(e->src_output() == 0) in CloneSmallHostConstantInputs()
128 << e->src_output(); in CloneSmallHostConstantInputs()
/external/tensorflow/tensorflow/core/debug/
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()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc158 if (e->src_output() == Graph::kControlSlot) { in Initialize()
160 } else if (e->src_output() != 0) { in Initialize()
161 return errors::Internal("Invalid output index ", e->src_output(), in Initialize()
177 if (e->src_output() == Graph::kControlSlot) { in Initialize()
179 } else if (e->src_output() != 0) { in Initialize()
180 return errors::Internal("Invalid output index ", e->src_output(), in Initialize()
201 kernel_state.output_locations[e->src_output()].push_back( in Initialize()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc57 if (edge->src_output() == old_src.second) { in ReplaceEdgeSources()
257 {edge->src(), edge->src_output()})) { in AddConversionMapping()
269 {edge->src(), edge->src_output()})); in AddConversionMapping()
304 TensorDesc output({ret_edge->src(), ret_edge->src_output()}); in ConvertOutput()
559 {edge->src(), edge->src_output()}, visited); in AddUnstackedTensorMappingsHelper()
580 {edge->src(), edge->src_output()})) { in AddUnstackedTensorMappingsHelper()
603 AddUnstackedTensorMappingsHelper({in_edge->src(), in_edge->src_output()}, in AddUnstackedTensorMappings()

1234