Home
last modified time | relevance | path

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

123

/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()
223 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
224 BaseType(edges[1]->src()->output_type(edges[1]->src_output()))); in InsertInputConversionNode()
225 CHECK_EQ(BaseType(edges[0]->src()->output_type(edges[0]->src_output())), in InsertInputConversionNode()
238 .Input(edges[0]->src(), edges[0]->src_output()) in InsertInputConversionNode()
239 .Input(edges[1]->src(), edges[1]->src_output()) in InsertInputConversionNode()
240 .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()
287 BackpropAlongEdge(y_grad_node_outputs_[i], {e->src(), e->src_output()}); in InitBackprop()
367 BackpropZerosAlongEdge({e->src(), e->src_output()}); in Compute()
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()
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/core/common_runtime/
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()
140 .Attr("tensor_type", edge->src()->output_type(edge->src_output())) in Recv()
181 const bool has_ref = IsRefType(e->src()->output_type(e->src_output())); in EnsureMemoryTypes()
183 Endpoint key{e->src()->id(), e->src_output()}; in EnsureMemoryTypes()
Dshape_refiner.cc233 input_shapes[e->dst_input()] = c->output(e->src_output()); in AddNode()
235 const auto* in_v = c->output_handle_shapes_and_types(e->src_output()); in AddNode()
237 DataType input_type = e->src()->output_type(e->src_output()); in AddNode()
330 int src_output = e->src_output(); in UpdateNode() local
344 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()
351 if (node_context->RelaxInput(dst_input, c->output(src_output))) { in UpdateNode()
366 if (e->src()->output_type(src_output) == DT_RESOURCE) { in UpdateNode()
367 auto* outputs = c->output_handle_shapes_and_types(src_output); in UpdateNode()
422 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
459 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()
[all …]
Dlower_if_op.cc67 Status AddInput(Node* src, int src_output);
147 Status CondBuilder::AddInput(Node* src, int src_output) { in AddInput() argument
161 .Input(src, src_output) in AddInput()
178 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output())); in AddInputs()
232 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()
451 {{added_nodes.second[out_edge->src_output()], 0}, in GetConstantGraph()
517 if (out_edge->src_output() == tensor.second) { in ReplaceTensorWithConstant()
Daccumulate_n_optimizer.cc130 .Input(data_edges[0]->src(), data_edges[0]->src_output()) in rewriteNode()
154 data_edges[i]->src_output()) // value: T in rewriteNode()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc128 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()
135 *val = OutputTensor(val_edge->src(), val_edge->src_output()); in GetSwitchValue()
198 explicit CondArgNode(Node* src, int src_output) in CondArgNode()
199 : src(src), src_output(src_output) {} in CondArgNode()
202 return absl::StrCat("src=", src->name(), ":", src_output, in ToString()
207 int src_output; member
415 std::pair<Node*, int> key = std::make_pair(e->src(), e->src_output()); in BuildArgumentNodes()
426 DataType dtype = cond_arg_node.src->output_type(cond_arg_node.src_output); in BuildArgumentNodes()
440 int branch_index = e->src_output(); in BuildArgumentNodes()
496 int src_output = edge->src_output(); in AddSwitchNodeAlongEdge() local
[all …]
Dfunctionalize_while.cc124 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() in CopySubgraph() local
126 : e->src_output(); in CopySubgraph()
128 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input()); in CopySubgraph()
323 graph->AddEdge(e->src(), e->src_output(), new_arg.enter, in FunctionalizeLoop()
443 if (edge->src_output() == 0) { in FunctionalizeLoop()
540 in_edge->src()->name(), in_edge->src_output(), arg_types[i])); in FunctionalizeLoop()
555 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
Dconst_analysis.cc81 status = pred->src()->input_edge(pred->src_output(), &pred); in BackwardsConstAnalysis()
111 status = edge->src()->input_edge(edge->src_output(), &edge); in BackwardsConstAnalysis()
/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.cc141 int src_output = e->src_output(), dst_input = e->dst_input(); in PreprocessDataEdgesBetweenOutsideCompilations() local
146 absl::StrCat(src->name(), "_oc_to_oc_placeholder_", src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
147 auto placeholder_index = std::make_pair(src->name(), src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
152 placeholder_builder.Attr("dtype", src->output_type(src_output)); in PreprocessDataEdgesBetweenOutsideCompilations()
162 src_output); in PreprocessDataEdgesBetweenOutsideCompilations()
228 data_edges.push_back({e->dst(), e->src_output(), e->dst_input()}); in PostprocessDataEdgesBetweenOutsideCompilations()
Dencapsulate_xla_computations_pass.cc227 e->src()->name(), ":", e->src_output(), " -> ", e->dst()->name(), ":", in Encapsulate()
283 data_inputs[i] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
291 data_inputs[pos] = {edge->src(), edge->src_output()}; in BuildXlaLaunchOps()
304 TF_RET_CHECK(le->src_output() < num_outputs); in BuildXlaLaunchOps()
313 data_outputs[le->src_output()].push_back( in BuildXlaLaunchOps()
316 output_types[le->src_output()] = output_node->input_type(0); in BuildXlaLaunchOps()
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()
Dclone_constants_for_better_clustering.cc49 g->AddEdge(e->src(), e->src_output(), new_in, e->dst_input()); in CloneNode()
111 TF_RET_CHECK(e->src_output() == 0) in CloneSmallHostConstantInputs()
114 << e->src_output(); in CloneSmallHostConstantInputs()
Dbuild_xla_ops_pass.cc55 g->AddEdge(new_node, edge->src_output(), edge->dst(), edge->dst_input()); in MoveOutgoingEdges()
89 int oidx = e->src_output(); in MergeOutgoingDataEdges()
157 return Output(e->src(), e->src_output()); in IncomingEdgeAsOutput()
244 args[e->dst_input()] = Output(e->src(), e->src_output()); in ReplaceFunctionCallWithPartionedCall()
265 g->AddEdge(call.operation.node(), e->src_output(), e->dst(), in ReplaceFunctionCallWithPartionedCall()
Dpartially_decluster_pass.cc72 if (output_mtypes[e->src_output()] == DEVICE_MEMORY) { in FindNodesToDecluster()
135 graph->AddEdge(in_edge->src(), in_edge->src_output(), cloned_node, in PartiallyDeclusterNode()
140 graph->AddEdge(cloned_node, out_edge_to_clone->src_output(), in PartiallyDeclusterNode()
Dencapsulate_subgraphs_pass.cc720 return results_.at(OutputTensor(edge->src(), edge->src_output())); in GetResultIndexForEdge()
732 .inputs.at(OutputTensor(edge->src(), edge->src_output())); in GetRecvAtHostSlot()
774 int src_slot = edge->src_output(); in RecordArg()
810 int src_slot = edge->src_output(); in RecordResult()
854 OutputTensor(edge->src(), edge->src_output()), input_index); in RecordOutsideCompilationInputOrControl()
868 .emplace(OutputTensor(edge->src(), edge->src_output()), in RecordOutsideCompilationOutputOrControl()
1457 g->AddEdge(src_image, edge->src_output(), dst_image, edge->dst_input()); in CopySubgraphEdges()
1465 DataType dtype = edge->src()->output_type(edge->src_output()); in CopySubgraphEdges()
1470 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
1500 edge->src()->name(), ":", edge->src_output()); in CopySubgraphEdges()
[all …]
Dextract_outside_compilation_pass.cc138 {edge->dst(), edge->src_output(), edge->dst_input()}); in ReplaceArgNodesWithRecvAtHostNode()
221 NodeDefBuilder::NodeOut{edge->src()->name(), edge->src_output(), in BuildSendFromHostNode()
222 edge->src()->output_type(edge->src_output())}; in BuildSendFromHostNode()
250 if (edge->src_output() == Graph::kControlSlot) { in ReplaceRetNodesWithSendFromHostNode()
253 g->AddEdge(edge->src(), edge->src_output(), send_from_host_node, index); in ReplaceRetNodesWithSendFromHostNode()
280 const PartialTensorShape shape = shapes[e->src_output()]; in GetInferredInputShapes()
329 e->src()->name(), e->src_output(), input_dtypes[e->dst_input()]}; in BuildXlaHostComputeNodeDef()
345 DataType dtype = e->src()->output_type(e->src_output()); in ValidateOutsideCompilationCallNode()
549 host_graph.AddEdge(node_map[e->src()], e->src_output(), copy, in ConstructHostGraph()
663 main_graph->AddEdge(node_map[e->src()], e->src_output(), copy, in ExpandHostGraphIntoMainGraph()
[all …]
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc110 const int src_output = edge->src_output(); in InsertNodes() local
111 if (output_slot_to_edges.find(src_output) == output_slot_to_edges.end()) { in InsertNodes()
112 output_slot_to_edges[src_output] = {edge}; in InsertNodes()
114 output_slot_to_edges[src_output].push_back(edge); in InsertNodes()
268 const string& src_node_name, const int src_output, const DataType src_dt, in CreateCopyNode() argument
277 const string copy_node_name = GetCopyNodeName(src_node_name, src_output); in CreateCopyNode()
301 .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()
566 {edge->src(), edge->src_output()}, visited); in AddUnstackedTensorMappingsHelper()
587 {edge->src(), edge->src_output()})) { in AddUnstackedTensorMappingsHelper()
610 AddUnstackedTensorMappingsHelper({in_edge->src(), in_edge->src_output()}, in AddUnstackedTensorMappings()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc80 input_edge->src_output()); in IsTensorRTCandidate()
284 const string s = StrCat(input_node->name(), ":", edge->src_output()); in GetEngineInfo()
293 input_node->name(), input_node->id(), edge->src_output(), node_name, in GetEngineInfo()
316 const string s = StrCat(node_name, ":", edge->src_output()); in GetEngineInfo()
326 node_name, node_id, edge->src_output(), /*input_edge=*/false, port); in GetEngineInfo()
682 NodeDefBuilder::NodeOut nout(edge->src()->name(), edge->src_output(), in RegisterSegmentFunctionToFunctionLibrary()
683 edge->src()->output_type(edge->src_output())); in RegisterSegmentFunctionToFunctionLibrary()
701 << edge->src_output() << " - > " << node_ret->name() << ":" << 0; in RegisterSegmentFunctionToFunctionLibrary()
702 sgraph.AddEdge(edge->src(), edge->src_output(), node_ret, 0); in RegisterSegmentFunctionToFunctionLibrary()
703 s = sgraph.UpdateEdge(edge->src(), edge->src_output(), node_ret, 0); in RegisterSegmentFunctionToFunctionLibrary()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dnode_def.proto21 // Each input is "node:src_output" with "node" being a string name and
22 // "src_output" indicating which output tensor to use from "node". If
23 // "src_output" is 0 the ":0" suffix can be omitted. Regular inputs
/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()

123