Home
last modified time | relevance | path

Searched refs:src_output (Results 1 – 25 of 76) 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.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
423 std::pair<Node*, int> key = std::make_pair(e->src(), e->src_output()); in BuildArgumentNodes()
434 DataType dtype = cond_arg_node.src->output_type(cond_arg_node.src_output); in BuildArgumentNodes()
448 int branch_index = e->src_output(); in BuildArgumentNodes()
504 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()
244 graph->AddEdge(e->src(), e->src_output(), new_arg.enter, in FunctionalizeLoop()
364 if (edge->src_output() == 0) { in FunctionalizeLoop()
457 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
472 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc179 NodeDefBuilder::NodeOut input(edge->src()->name(), edge->src_output(), in AddInputIdentity()
187 << edge->src_output() << " -> " << edge->dst()->name() << ":" in AddInputIdentity()
196 graph->AddEdge(edge->src(), edge->src_output(), identity_node, 0); in AddInputIdentity()
214 auto add_identity = [&](int src_output, const string& identity_name, in AddOutputIdentities()
217 builder.Attr("T", node->output_type(src_output)); in AddOutputIdentities()
218 NodeDefBuilder::NodeOut input(node->name(), src_output, in AddOutputIdentities()
219 node->output_type(src_output)); in AddOutputIdentities()
230 graph->AddEdge(node, src_output, *identity_node, 0); in AddOutputIdentities()
246 output_used[edge->src_output()] = true; in AddOutputIdentities()
250 int src_output = edge->src_output(); in AddOutputIdentities() local
[all …]
Dmemory_types.cc82 MemoryType sm = gtl::FindWithDefault(out, {e->src()->id(), e->src_output()}, in ProcessMemoryTypes()
86 VLOG(1) << e->src()->id() << ":" << e->src_output() << " -> " in ProcessMemoryTypes()
102 e->src_output(), " and ", e->dst()->id(), ":", in ValidateMemoryTypes()
126 .Input(edge->src(), edge->src_output()) in Send()
141 .Attr("tensor_type", edge->src()->output_type(edge->src_output())) in Recv()
182 const bool has_ref = IsRefType(e->src()->output_type(e->src_output())); in EnsureMemoryTypes()
184 Endpoint key{e->src()->id(), e->src_output()}; in EnsureMemoryTypes()
Dshape_refiner.cc218 ic->SetInput(e->dst_input(), input_ic->output(e->src_output())); in AddNode()
221 input_ic->output_handle_shapes_and_types(e->src_output()); in AddNode()
223 DataType input_type = e->src()->output_type(e->src_output()); in AddNode()
303 int src_output = e->src_output(); in UpdateNode() local
317 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()
324 if (node_context->RelaxInput(dst_input, c->output(src_output))) { in UpdateNode()
339 if (e->src()->output_type(src_output) == DT_RESOURCE) { in UpdateNode()
340 auto* outputs = c->output_handle_shapes_and_types(src_output); in UpdateNode()
395 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
436 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
[all …]
Dlower_if_op.cc67 Status AddInput(Node* src, int src_output);
158 Status CondBuilder::AddInput(Node* src, int src_output) { in AddInput() argument
172 .Input(src, src_output) in AddInput()
189 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output())); in AddInputs()
246 graph_->AddEdge(merges[e->src_output()], 0, e->dst(), e->dst_input()); in AddOutputs()
Dlower_case_op.cc65 Status AddInput(Node* src, int src_output);
147 Status CaseBuilder::AddInput(Node* src, int src_output) { in AddInput() argument
161 .Input(src, src_output) in AddInput()
180 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output())); in AddInputs()
248 graph_->AddEdge(merges[e->src_output()], 0, e->dst(), e->dst_input()); in AddOutputs()
Dconstant_folding.cc74 CHECK_GT(known_shape.size(), in->src_output()) << known_shape_iter->first; in ReadPartialShapesFromShapeMap()
77 (*input_shapes)[in->dst_input()] = known_shape[in->src_output()]; in ReadPartialShapesFromShapeMap()
371 constant_graph->AddEdge(it->second[0], in_edge->src_output(), added[0], in AddNodeToConstantGraph()
378 constant_graph->AddEdge(it->second[in_edge->src_output()], 0, added[0], in AddNodeToConstantGraph()
443 {{added_nodes.second[0], out_edge->src_output()}, in GetConstantGraph()
444 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
451 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
452 {added_nodes.first, out_edge->src_output()}}); in GetConstantGraph()
517 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
Daccumulate_n_optimizer.cc162 .Input(data_edges[0]->src(), data_edges[0]->src_output()) in RewriteIntoTempVariable()
186 data_edges[i]->src_output()) // value: T in RewriteIntoTempVariable()
251 data_inputs.emplace_back(in_edge->src(), in_edge->src_output()); in RewriteIntoAddN()
/external/tensorflow/tensorflow/core/graph/
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 …]
Dgradients.cc114 n_inputs[e->dst_input()] = {e->src(), e->src_output()}; in AddSymGrad()
295 BackpropAlongEdge(y_grad_node_outputs_[i], {e->src(), e->src_output()}); in InitBackprop()
375 BackpropZerosAlongEdge({e->src(), e->src_output()}); in Compute()
386 graph_->AddEdge(e->src(), e->src_output(), grad, e->dst_input()); in Compute()
395 BackpropAlongEdge({grad, e->dst_input()}, {e->src(), e->src_output()}); in Compute()
Dsubgraph.cc93 if (e->src_output() == id.second) { in FeedInputs()
95 } else if (e->src_output() == Graph::kControlSlot && in FeedInputs()
108 if (e->src_output() == id.second) { in FeedInputs()
111 CHECK_EQ(Graph::kControlSlot, e->src_output()); in FeedInputs()
/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()
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()
329 g->AddEdge(call.operation.node(), e->src_output(), e->dst(), in ReplaceFunctionCallWithPartitionedCall()
431 if (e->src()->output_type(e->src_output()) == DT_INT32) { in PredicateInt32Inputs()
434 if (source_output_mem_types[e->src_output()] == DEVICE_MEMORY) { in PredicateInt32Inputs()
435 int32_inputs.push_back(Output(e->src(), e->src_output())); in PredicateInt32Inputs()
Dencapsulate_xla_computations_pass.cc228 e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":", in Encapsulate()
284 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
292 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
305 TF_RET_CHECK(le->src_output() < num_outputs); in BuildXlaLaunchOps()
314 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()
317 output_types[le->src_output()] = output_node->input_type(0); in BuildXlaLaunchOps()
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()
Dencapsulate_subgraphs_pass.cc442 return results_.at(OutputTensor(edge->src(), edge->src_output())); in GetResultIndexForEdge()
472 int src_slot = edge->src_output(); in RecordArg()
517 int src_slot = edge->src_output(); in RecordResult()
727 g->AddEdge(src_image, edge->src_output(), dst_image, edge->dst_input()); in CopySubgraphEdges()
735 DataType dtype = edge->src()->output_type(edge->src_output()); in CopySubgraphEdges()
740 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
763 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
879 return edge->src_output(); in FindOutputSlotOfEdgeSrc()
941 int src_output = FindOutputSlotOfEdgeSrc(src_func_id, dst_func_id, edge); in CopyEdgeToOutputGraph() local
947 ->emplace(OutputTensor(src_image, src_output), in CopyEdgeToOutputGraph()
[all …]
Dpartially_decluster_pass.cc84 if (output_mtypes[e->src_output()] == DEVICE_MEMORY) { in FindNodesToDecluster()
147 graph->AddEdge(in_edge->src(), in_edge->src_output(), cloned_node, in PartiallyDeclusterNode()
152 graph->AddEdge(cloned_node, out_edge_to_clone->src_output(), in PartiallyDeclusterNode()
/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/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()
/external/tensorflow/tensorflow/cc/framework/
Dops.cc33 return Output(e->src(), e->src_output()); in input()
58 inputs[e->dst_input()] = std::make_pair(e->src(), e->src_output()); in GetInputs()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc135 if (e->src_output() == Graph::kControlSlot) { in Initialize()
137 } else if (e->src_output() != 0) { in Initialize()
138 return errors::Internal("Invalid output index ", e->src_output(), in Initialize()
156 kernel_state.output_locations[e->src_output()].push_back( in Initialize()

1234