| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 51 std::vector<NodeBuilder::NodeOut> out_nodes; in ReplaceReduce() local
53 out_nodes.emplace_back(edge->dst(), edge->dst_input()); in ReplaceReduce()
82 for (const auto& out_node : out_nodes) { in ReplaceReduce()
94 for (const auto& out_node : out_nodes) { in ReplaceReduce()
149 for (const auto& out_nodes : out_nodes_map) { in ReplaceBroadcast() local
150 for (const auto& out_node : out_nodes) { in ReplaceBroadcast()
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
| D | threads.cpp | 70 void CalculateProcessorTopology(CPUNumaNodes& out_nodes, uint32_t& out_numThreadsPerProcGroup) in CalculateProcessorTopology() argument
72 out_nodes.clear(); in CalculateProcessorTopology()
142 if (out_nodes.size() <= numaId) in CalculateProcessorTopology()
144 out_nodes.resize(numaId + 1); in CalculateProcessorTopology()
146 auto& numaNode = out_nodes[numaId]; in CalculateProcessorTopology()
199 if (physId + 1 > out_nodes.size()) in CalculateProcessorTopology()
200 out_nodes.resize(physId + 1); in CalculateProcessorTopology()
201 auto& numaNode = out_nodes[physId]; in CalculateProcessorTopology()
213 for (auto& node : out_nodes) in CalculateProcessorTopology()
242 out_nodes.resize(numPhysicalIds); in CalculateProcessorTopology()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 217 for (const Node* consumer : placeholder_node->out_nodes()) { in TEST()
228 for (const Node* consumer : enter_node->out_nodes()) { in TEST()
248 for (const Node* consumer : merge_node->out_nodes()) { in TEST()
258 for (const Node* consumer : switch_node->out_nodes()) { in TEST()
338 for (const Node* consumer : placeholder_node->out_nodes()) { in TEST()
349 for (const Node* consumer : enter_node->out_nodes()) { in TEST()
369 for (const Node* consumer : merge_node->out_nodes()) { in TEST()
379 for (const Node* consumer : switch_node->out_nodes()) { in TEST()
|
| D | immutable_executor_state.cc | 142 for (const Node* consumer : n->out_nodes()) { in Initialize()
|
| /external/autotest/client/cros/multimedia/ |
| D | usb_facade_native.py | 110 out_nodes, in_nodes = cras_utils.get_plugged_node_types()
112 out_nodes, in_nodes)
113 return 'USB' in out_nodes and 'USB' in in_nodes
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | variable_merger_pass.cc | 146 for (Node* m : graph->source_node()->out_nodes()) { in Run()
151 for (Node* n : m->out_nodes()) { in Run()
|
| D | host_training_loop_optimization_util.cc | 127 for (Node* node : compile_node->out_nodes()) { in ExtractExecuteNodeInfo()
300 for (auto n : loop_cond_node.out_nodes()) { in GetLoopSwitchNode()
351 for (const auto out_node : loop_switch_node->out_nodes()) { in AddNoOpAfterLastIteration()
|
| D | distributed_tpu_rewrite_pass.cc | 1089 for (Node* out : unpadded_input->out_nodes()) { in SetPaddingNodesDevices()
1099 for (Node* out : n->out_nodes()) { in SetPaddingNodesDevices()
1250 for (const Node* connected_node : input_node.out_nodes()) in FindNodesMaybeContainingShardingInfo()
2551 for (Node* node : key_placeholder_node->out_nodes()) { in ConnectHostComputeNodes()
4086 for (Node* successor : replicate_node.out_nodes()) { in BuildSequencingNodes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 99 std::vector<SimpleNode*> out_nodes() const { in out_nodes() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleNode
260 : [](const SimpleNode* n) { return n->out_nodes(); }; in StableDFS()
997 : node->out_nodes()) { in SegmentGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 146 gtl::iterator_range<NeighborIter> out_nodes() const;
|
| D | graph.cc | 197 gtl::iterator_range<NeighborIter> Node::out_nodes() const { in out_nodes() function in tensorflow::Node
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | tfcompile.bzl | 115 # First run tfcompile to generate the list of out_nodes.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 1009 for (auto n : replacee->out_nodes()) in PropagateUpdatedState()
1022 for (auto out : n->out_nodes()) in PropagateUpdatedState()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 760 for (const Node* out : node->out_nodes()) { in ClusterizeNodes()
|