| /external/tensorflow/tensorflow/core/common_runtime/ |
| parallel_concat_optimizer.cc | 82 for (const Edge* input_edge : n->in_edges()) {
83 if (input_edge->IsControlEdge()) {
84 g->AddControlEdge(input_edge->src(), start);
93 .Input(input_edge->src(), input_edge->src_output())
|
| accumulate_n_optimizer.cc | 100 for (const Edge* input_edge : n->in_edges()) {
101 if (input_edge->IsControlEdge()) {
102 control_edges.push_back(input_edge);
104 data_edges.push_back(input_edge);
|
| shape_refiner.cc | 411 const Edge* input_edge; local
412 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge));
415 const Node* src = input_edge->src();
440 input_edge->src(), &subgraph, &is_constant_graph, &const_inputs));
445 strings::StrCat(input_edge->src()->name(), ":", input_edge->src_output());
710 const Edge* input_edge; local
711 TF_RETURN_IF_ERROR(node->input_edge(dst_idx, &input_edge));
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 503 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge));
519 TF_RETURN_IF_ERROR(arg.enter->input_edge(0, &in_edge));
652 const std::vector<Node*>& merge_nodes, int input_edge,
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_test.cc | 226 TF_ASSERT_OK(c->input_edge(0, &e));
244 TF_ASSERT_OK(t->input_edge(1, &e));
271 TF_ASSERT_OK(c->input_edge(0, &a_new_c_edge));
276 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge));
282 TF_ASSERT_OK(c->input_edge(0, &b_new_c_edge));
|
| graph.cc | 195 Status Node::input_edge(int idx, const Edge** e) const { function in class:tensorflow::Node
197 return errors::InvalidArgument("Invalid input_edge index: ", idx, ", Node ",
248 TF_RETURN_IF_ERROR(input_edge(idx, &e));
|
| graph.h | 175 Status input_edge(int idx, const Edge** e) const;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| graph_to_functiondef.cc | 155 TF_CHECK_OK(node->input_edge(0, &edge));
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_rewrite.cc | 143 TF_RETURN_IF_ERROR(node->input_edge(0, &in_edge));
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 725 TF_CHECK_OK(node.input_edge(1, &edge));
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.cc | [all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
