| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_rewrite.cc | 144 Node* in_node = in_edge->src(); local
150 graph->AddControlEdge(in_node, out_node.node);
152 graph->AddEdge(in_node, in_index, out_node.node, out_node.index);
168 NodeBuilder::NodeOut in_node; local
174 in_node = NodeBuilder::NodeOut(edge->src(), edge->src_output());
175 send_builder.Input(in_node);
193 graph->AddEdge(in_node.node, in_node.index, out_node.node,
203 string shape_name = strings::StrCat(in_node.node->name(), "/Shape");
207 shape_builder.Input(in_node).Attr("out_type", DT_INT32).Attr("T", dtype)
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session.cc | 1048 const Node* in_node = in_edge->src(); local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 261 const Node* in_node; local
262 TF_CHECK_OK(node.input_node(0, &in_node));
263 return IsQueue(*in_node);
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_session.cc | 875 const Node* in_node = in_edge->src(); local
876 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) {
877 return errors::InvalidArgument("Fetch ", in_node->name(), ":",
882 if (!visited[in_node->id()]) {
883 visited[in_node->id()] = true;
884 stack.push_back(in_node);
[all...] |
