Searched refs:in_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 146 Node* in_node = in_edge->src(); in ReplaceBroadcast() local
152 graph->AddControlEdge(in_node, out_node.node); in ReplaceBroadcast()
154 graph->AddEdge(in_node, in_index, out_node.node, out_node.index); in ReplaceBroadcast()
170 NodeBuilder::NodeOut in_node; in ReplaceBroadcast() local
176 in_node = NodeBuilder::NodeOut(edge->src(), edge->src_output()); in ReplaceBroadcast()
177 send_builder.Input(in_node); in ReplaceBroadcast()
195 graph->AddEdge(in_node.node, in_node.index, out_node.node, in ReplaceBroadcast()
205 string shape_name = strings::StrCat(in_node.node->name(), "/Shape"); in ReplaceBroadcast()
209 shape_builder.Input(in_node).Attr("out_type", DT_INT32).Attr("T", dtype); in ReplaceBroadcast()
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 229 for (int in_node = 0; in_node < num_inputs; in_node++) { in av1_nn_predict_sse3() local
230 __m128 input = _mm_load1_ps(&input_nodes[in_node]); in av1_nn_predict_sse3()
232 _mm_load1_ps(&layer_weights[num_inputs * out + in_node]); in av1_nn_predict_sse3()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1275 const Node* in_node = in_edge->src(); in CheckFetch() local
1276 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetch()
1277 return errors::InvalidArgument("Fetch ", in_node->name(), ":", in CheckFetch()
1282 if (!visited[in_node->id()]) { in CheckFetch()
1283 visited[in_node->id()] = true; in CheckFetch()
1284 stack.push_back(in_node); in CheckFetch()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 1077 const Node* in_node = in_edge->src(); in CheckFetches() local
1078 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetches()
1079 return errors::InvalidArgument("Fetch ", in_node->name(), ":", in CheckFetches()
1084 if (!visited[in_node->id()]) { in CheckFetches()
1085 visited[in_node->id()] = true; in CheckFetches()
1086 stack.push_back(in_node); in CheckFetches()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 1550 auto* in_node = context.graph_view->GetNode("identity_n"); in TEST_F() local
1551 ASSERT_NE(in_node, nullptr); in TEST_F()
1552 ASSERT_EQ(in_node->NumRegularFanins(), 4); in TEST_F()
1553 VerifyRegularFaninMatch(in_node, 0, "conv2d_1", 0); in TEST_F()
1554 VerifyRegularFaninMatch(in_node, 1, input_transpose_node2->GetName(), 0); in TEST_F()
1555 VerifyRegularFaninMatch(in_node, 2, "a", 0); in TEST_F()
1556 VerifyRegularFaninMatch(in_node, 3, "b", 0); in TEST_F()
1566 VerifyRegularFaninMatch(output_transpose_node2, 0, in_node->GetName(), 1); in TEST_F()
1579 VerifyRegularFaninMatch(conv2d_1_output_node, 0, in_node->GetName(), 0); in TEST_F()
1590 VerifyRegularFaninMatch(a_output_node, 0, in_node->GetName(), 2); in TEST_F()
[all …]
|