| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_launch_ops_pass.cc | 95 graph->AddEdge(edge->src(), edge->src_output(), launch_node,
105 int src_output = edge->src_output(); local
112 graph->AddEdge(launch_node, src_output, dst, dst_input);
|
| encapsulate_subgraphs_pass.cc | 608 return results_.at(NodeSlot(edge->src(), edge->src_output()));
620 .inputs.at(NodeSlot(edge->src(), edge->src_output()));
656 int src_slot = edge->src_output();
691 int src_slot = edge->src_output();
725 outside_subgraph.inputs.emplace(NodeSlot(edge->src(), edge->src_output()),
1541 int src_output = local
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 110 const int src_output = edge->src_output(); local
111 if (output_slot_to_edges.find(src_output) == output_slot_to_edges.end()) {
112 output_slot_to_edges[src_output] = {edge};
114 output_slot_to_edges[src_output].push_back(edge);
268 const string& src_node_name, const int src_output, const DataType src_dt,
277 const string copy_node_name = GetCopyNodeName(src_node_name, src_output);
301 .Input(src_node_name, src_output, src_dt)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 232 input_shapes[e->dst_input()] = c->output(e->src_output());
235 if (e->src()->output_type(e->src_output()) == DT_RESOURCE) {
236 const auto* in_v = c->output_handle_shapes_and_types(e->src_output());
325 int src_output = e->src_output(); local
339 if (node_context->MergeInput(dst_input, c->output(src_output))) {
346 if (node_context->RelaxInput(dst_input, c->output(src_output))) {
356 if (e->src()->output_type(src_output) == DT_RESOURCE) {
357 auto* outputs = c->output_handle_shapes_and_types(src_output);
445 strings::StrCat(input_edge->src()->name(), ":", input_edge->src_output());
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 121 // we rewrite the src_output of the corresponding edge to be 0.
150 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() local
152 : e->src_output();
154 output->AddEdge(src_copy, src_output, dst_copy, e->dst_input());
316 graph->AddEdge(e->src(), e->src_output(), new_arg.enter,
433 if (edge->src_output() == 0) {
508 in_edge->src()->name(), in_edge->src_output(), arg_types[i]));
523 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i);
1307 int src_output = local
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph.h | 304 int src_output() const { return src_output_; } function in class:tensorflow::Edge
|
