Home
last modified time | relevance | path

Searched refs:dst_input (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc113 int dst_input, dst_node_id; in PreprocessDataEdgesBetweenOutsideCompilations() member
128 edges.push_back(EdgeInfo{e->dst_input(), e->dst()->id()}); in PreprocessDataEdgesBetweenOutsideCompilations()
139 TF_RETURN_IF_ERROR(dst->input_edge(edges[i].dst_input, &e)); in PreprocessDataEdgesBetweenOutsideCompilations()
141 int src_output = e->src_output(), dst_input = e->dst_input(); in PreprocessDataEdgesBetweenOutsideCompilations() local
172 g->AddEdge(placeholder_node, 0, dst, dst_input); in PreprocessDataEdgesBetweenOutsideCompilations()
176 *new_def.mutable_input(dst_input) = placeholder_node->name(); in PreprocessDataEdgesBetweenOutsideCompilations()
228 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
238 int dst_input = data_edges[i].dst_input; in PostprocessDataEdgesBetweenOutsideCompilations() local
239 *new_def.mutable_input(dst_input) = in PostprocessDataEdgesBetweenOutsideCompilations()
244 TF_RETURN_IF_ERROR(replace_node->input_edge(dst_input, &edge_to_replace)); in PostprocessDataEdgesBetweenOutsideCompilations()
[all …]
Dclone_constants_for_better_clustering.cc49 g->AddEdge(e->src(), e->src_output(), new_in, e->dst_input()); in CloneNode()
110 int dst_input = e->dst_input(); in CloneSmallHostConstantInputs() local
116 g->AddEdge(input_cloned, 0, n, dst_input); in CloneSmallHostConstantInputs()
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()
Dextract_outside_compilation_pass.cc138 {edge->dst(), edge->src_output(), edge->dst_input()}); in ReplaceArgNodesWithRecvAtHostNode()
142 if (edge.dst_input == Graph::kControlSlot) { in ReplaceArgNodesWithRecvAtHostNode()
145 g->AddEdge(recv_at_host_node, index, edge.dst, edge.dst_input); in ReplaceArgNodesWithRecvAtHostNode()
152 if (edge.dst_input == Graph::kControlSlot) { in ReplaceArgNodesWithRecvAtHostNode()
158 *new_def.mutable_input(edge.dst_input) = in ReplaceArgNodesWithRecvAtHostNode()
285 results[e->dst_input()] = shape; in GetInferredInputShapes()
325 if (e->dst_input() < 0 || e->dst_input() >= input_dtypes.size()) { in BuildXlaHostComputeNodeDef()
326 return errors::Internal("Invalid dst_input: ", e->dst_input()); in BuildXlaHostComputeNodeDef()
328 inputs[e->dst_input()] = NodeDefBuilder::NodeOut{ in BuildXlaHostComputeNodeDef()
329 e->src()->name(), e->src_output(), input_dtypes[e->dst_input()]}; in BuildXlaHostComputeNodeDef()
[all …]
Dpartially_decluster_pass.cc86 dst_input_mtypes[e->dst_input()] == HOST_MEMORY; in FindNodesToDecluster()
136 in_edge->dst_input()); in PartiallyDeclusterNode()
141 out_edge_to_clone->dst(), out_edge_to_clone->dst_input()); in PartiallyDeclusterNode()
Dencapsulate_subgraphs_pass.cc716 return args_by_dst_.at(InputTensor(edge->dst(), edge->dst_input())); in GetArgIndexForEdge()
744 .outputs_by_dst.at(InputTensor(edge->dst(), edge->dst_input())); in GetSendFromHostSlot()
785 DataType dtype = edge->dst()->input_type(edge->dst_input()); in RecordArg()
799 int dst_slot = edge->dst_input(); in RecordArg()
865 DataType dtype = edge->dst()->input_type(edge->dst_input()); in RecordOutsideCompilationOutputOrControl()
874 ->outputs_by_dst[InputTensor(edge->dst(), edge->dst_input())] = in RecordOutsideCompilationOutputOrControl()
1457 g->AddEdge(src_image, edge->src_output(), dst_image, edge->dst_input()); in CopySubgraphEdges()
1495 DataType dtype = edge->dst()->input_type(edge->dst_input()); in CopySubgraphEdges()
1713 return edge->dst_input(); in FindOutputSlotOfEdgeDst()
1752 int dst_input = in CopyEdgeToOutputGraph() local
[all …]
Dbuild_xla_ops_pass.cc55 g->AddEdge(new_node, edge->src_output(), edge->dst(), edge->dst_input()); in MoveOutgoingEdges()
98 int dst_idx = e->dst_input(); in MergeOutgoingDataEdges()
244 args[e->dst_input()] = Output(e->src(), e->src_output()); in ReplaceFunctionCallWithPartionedCall()
266 e->dst_input()); in ReplaceFunctionCallWithPartionedCall()
Dincrease_dynamism_for_auto_jit_pass.cc290 int dst_input = e->dst_input(); in ReplaceTensorFlowSliceWithStaticShapedSlice() local
293 g->AddEdge(static_shaped_slice, src_output, dst, dst_input); in ReplaceTensorFlowSliceWithStaticShapedSlice()
Dshape_inference_helpers.h39 int dst_input; member
Dencapsulate_xla_computations_pass.cc84 return edge->dst()->input_type(edge->dst_input()); in EdgeType()
228 e->dst_input()); in Encapsulate()
314 {oe->dst(), oe->dst_input()}); in BuildXlaLaunchOps()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc226 "Input ", e->dst_input(), " ('", input->name(), "') for '", in AddNode()
231 DCHECK_GE(e->dst_input(), 0); in AddNode()
232 input_nodes[e->dst_input()] = input; in AddNode()
233 input_shapes[e->dst_input()] = c->output(e->src_output()); in AddNode()
239 input_handle_shapes_and_types[e->dst_input()].reset( in AddNode()
329 int dst_input = e->dst_input(); in UpdateNode() local
336 "Input ", dst_input, " ('", input->name(), "') for '", node->name(), in UpdateNode()
341 DCHECK_GE(dst_input, 0); in UpdateNode()
342 ShapeHandle existing_input = node_context->input(dst_input); in UpdateNode()
344 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()
[all …]
Dmemory_types.cc83 MemoryType dm = gtl::FindWithDefault(inp, {e->dst()->id(), e->dst_input()}, in ProcessMemoryTypes()
86 << e->dst()->id() << ":" << e->dst_input() << ": " << sm << " -> " in ProcessMemoryTypes()
102 e->dst_input(), " : from ", in ValidateMemoryTypes()
198 g->AddEdge(recv, 0, e->dst(), e->dst_input()); in EnsureMemoryTypes()
Dconstant_folding.cc75 DCHECK_GE(in->dst_input(), 0); in ReadPartialShapesFromShapeMap()
76 DCHECK_LT(in->dst_input(), input_shapes->size()); in ReadPartialShapesFromShapeMap()
77 (*input_shapes)[in->dst_input()] = known_shape[in->src_output()]; in ReadPartialShapesFromShapeMap()
372 in_edge->dst_input()); in AddNodeToConstantGraph()
379 in_edge->dst_input()); in AddNodeToConstantGraph()
545 graph->AddEdge(constant_node, 0, edge->dst(), edge->dst_input()); in ReplaceTensorWithConstant()
/external/tensorflow/tensorflow/core/graph/
Dmkl_tfconversion_pass.cc153 DataType dst_datatype = dst->input_type(e->dst_input()); in InsertConversionNodeOnEdge()
160 ":" + std::to_string(e->dst_input()) + in InsertConversionNodeOnEdge()
197 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 0, dst, e->dst_input())); in InsertConversionNodeOnEdge()
230 CHECK_EQ((edges[i]->dst_input() == i), true); in InsertInputConversionNode()
281 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 0, n, edges[0]->dst_input())); in InsertInputConversionNode()
282 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 1, n, edges[1]->dst_input())); in InsertInputConversionNode()
283 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 2, n, edges[2]->dst_input())); in InsertInputConversionNode()
284 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 3, n, edges[3]->dst_input())); in InsertInputConversionNode()
Dgraph.cc255 if (edge->dst_input() == idx) { in input_edge()
271 if (edge->dst_input() < 0 || edge->dst_input() >= num_inputs()) { in input_edges()
272 return errors::Internal("Invalid edge input number ", edge->dst_input()); in input_edges()
274 if ((*input_edges)[edge->dst_input()] != nullptr) { in input_edges()
276 edge->dst_input()); in input_edges()
278 (*input_edges)[edge->dst_input()] = edge; in input_edges()
557 if (e->dst_input() == index) return e; in FindEdge()
663 CHECK(inputs[edge->dst_input()] == nullptr) in ToGraphDefSubRange()
666 << edge->dst_input() << " and had pre-existing input edge " in ToGraphDefSubRange()
667 << inputs[edge->dst_input()]->src()->DebugString() << ":" in ToGraphDefSubRange()
[all …]
Doptimizer_cse.cc77 (*in)[e->dst_input()] = std::make_pair(e->src(), e->src_output()); in FillInputs()
214 g_->AddEdge(*candidate, e->src_output(), e->dst(), e->dst_input()); in Optimize()
Dmkl_layout_pass.cc880 e->dst_input() == 0) { in GetConv2DOrBiasAdd()
1036 e->dst_input() == 2 /* 3rd input of BackpropFilter */) { in GetConv2DBackpropFilterOrBiasAddGrad()
1051 e->dst_input() == 0 /* 1st input of BiasAddGrad */) { in GetConv2DBackpropFilterOrBiasAddGrad()
1119 e->dst_input() == kPermTensorIndex) { in CheckForTranspose()
1244 if (e->dst()->type_string() == csinfo_.lrn_grad && e->dst_input() == 2 && in LrnGradRewrite()
1286 e->dst_input() == 1 && in MaxpoolGradRewrite()
1567 (*in)[e->dst_input()] = std::make_pair(e->src(), e->src_output()); in FillInputs()
1738 e->dst_input() == kConv2DFilterInputSlotIdx in SetUpContiguousInputs()
1949 e->dst_input() == ws.bwd_slot) { in AddWorkSpaceEdgeIfNeeded()
1984 e->dst_input() == ws.bwd_slot) { in AddWorkSpaceEdgeIfNeeded()
[all …]
Dgradients.cc114 n_inputs[e->dst_input()] = {e->src(), e->src_output()}; in AddSymGrad()
378 graph_->AddEdge(e->src(), e->src_output(), grad, e->dst_input()); in Compute()
387 BackpropAlongEdge({grad, e->dst_input()}, {e->src(), e->src_output()}); in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc128 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input()); in CopySubgraph()
327 int dst_input = edges[i]->dst_input(); in FunctionalizeLoop() local
329 graph->AddEdge(new_arg.enter, 0, dst, dst_input); in FunctionalizeLoop()
351 edge->dst_input() == 1) { in FunctionalizeLoop()
406 TF_RETURN_IF_ERROR(arg.merge->input_node(1 - enter_merge->dst_input(), in FunctionalizeLoop()
418 if (edge->dst_input() == 0 && IsSwitch(edge->dst()) && in FunctionalizeLoop()
565 int dst_input = edge->dst_input(); in FunctionalizeLoop() local
568 if (dst_input == Graph::kControlSlot) { in FunctionalizeLoop()
571 graph->AddEdge(while_node, i, dst, dst_input); in FunctionalizeLoop()
Dtf2xla_util.cc136 int dst_node_id, dst_input; in ReplaceArgUsageWithConstNode() member
143 usages.push_back({e->dst()->id(), e->dst_input()}); in ReplaceArgUsageWithConstNode()
150 *replace_def.mutable_input(usages[i].dst_input) = const_node->name(); in ReplaceArgUsageWithConstNode()
155 replace_node->input_edge(usages[i].dst_input, &usage_edge)); in ReplaceArgUsageWithConstNode()
157 g->AddEdge(const_node, 0, replace_node, usages[i].dst_input); in ReplaceArgUsageWithConstNode()
649 edge->dst_input()); in RewriteAssociatedFunction()
653 edge->dst_input()); in RewriteAssociatedFunction()
727 {edge->dst(), edge->src_output(), edge->dst_input()}); in ReplaceNode()
736 in_edge->dst_input()); in ReplaceNode()
739 g->AddEdge(new_node, out_edge.src_output, out_edge.dst, out_edge.dst_input); in ReplaceNode()
Dfunctionalize_cond.cc453 int dst_input = IsMerge(e->dst()) ? 0 : e->dst_input(); in BuildArgumentNodes() local
454 bodies_[branch_index]->AddEdge(src_copy, 0, dst_copy, dst_input); in BuildArgumentNodes()
507 int dst_input = edge->dst_input(); in AddSwitchNodeAlongEdge() local
509 graph->AddEdge(switch_node, static_cast<int>(branch), dst, dst_input); in AddSwitchNodeAlongEdge()
683 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input()); in ExtractBodies()
852 int dst_input = edge->dst_input(); in AddOutputEdges() local
864 graph->AddEdge(if_node_, i, dst, dst_input); in AddOutputEdges()
1194 int dst_port = oe->dst_input(); in RemoveRedundantMerge()
1247 int dst_input = e->dst_input(); in RemoveRedundantSwitch() local
1270 dst_node, dst_input); in RemoveRedundantSwitch()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc761 for (int dst_input = 0; dst_input < ic->num_inputs(); ++dst_input) { in UpdateNode() local
762 const GraphView::InputPort port(node, dst_input); in UpdateNode()
770 "Input ", dst_input, " ('", src->name(), "') for '", node->name(), in UpdateNode()
786 input_tensors_as_shapes[dst_input] = in UpdateNode()
793 const_values[dst_input].FromProto(*tensor_proto)) { in UpdateNode()
794 input_tensors[dst_input] = &const_values[dst_input]; in UpdateNode()
795 ctx->input_tensor_protos[dst_input] = tensor_proto; in UpdateNode()
797 if (!ic->FullyDefined(input_tensors_as_shapes[dst_input])) { in UpdateNode()
822 MaybeTensorValueToShape(ic, const_values[dst_input], in UpdateNode()
823 &input_tensors_as_shapes[dst_input]); in UpdateNode()
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dops.cc32 if (e->dst_input() == i) { in input()
58 inputs[e->dst_input()] = std::make_pair(e->src(), e->src_output()); in GetInputs()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc182 if (in_edges.size() <= edge->dst_input()) { in GraphToFunctionDef()
183 in_edges.resize(edge->dst_input() + 1); in GraphToFunctionDef()
185 in_edges[edge->dst_input()] = edge; in GraphToFunctionDef()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc63 int dst_input() const { return dst_port_; } in dst_input() function in tensorflow::tensorrt::segment::SimpleEdge
171 auto edge = new SimpleEdge(i, src, e->src_output(), dst, e->dst_input(), in SimpleGraph()
386 graph->AddEdge(src, Graph::kControlSlot, e->dst(), e->dst_input()); in ContractEdge()
388 graph->AddEdge(src, 0 /* output index */, e->dst(), e->dst_input()); in ContractEdge()

123