Searched refs:in_node (Results 1 – 6 of 6) 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/deqp-deps/SPIRV-Tools/source/comp/ |
| D | huffman_codec.h | 329 Context(uint32_t in_node, uint64_t in_bits, size_t in_depth) in CreateEncodingTable()
330 : node(in_node), bits(in_bits), depth(in_depth) {} in CreateEncodingTable()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/comp/ |
| D | huffman_codec.h | 329 Context(uint32_t in_node, uint64_t in_bits, size_t in_depth) in CreateEncodingTable()
330 : node(in_node), bits(in_bits), depth(in_depth) {} in CreateEncodingTable()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1169 const Node* in_node = in_edge->src(); in CheckFetch() local
1170 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetch()
1171 return errors::InvalidArgument("Fetch ", in_node->name(), ":", in CheckFetch()
1176 if (!visited[in_node->id()]) { in CheckFetch()
1177 visited[in_node->id()] = true; in CheckFetch()
1178 stack.push_back(in_node); in CheckFetch()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 1020 const Node* in_node = in_edge->src(); in CheckFetches() local
1021 if (pending_feeds.count({in_node->name(), in_edge->src_output()}) > 0) { in CheckFetches()
1022 return errors::InvalidArgument("Fetch ", in_node->name(), ":", in CheckFetches()
1027 if (!visited[in_node->id()]) { in CheckFetches()
1028 visited[in_node->id()] = true; in CheckFetches()
1029 stack.push_back(in_node); in CheckFetches()
|