Home
last modified time | relevance | path

Searched refs:out_nodes (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc51 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/
Dthreads.cpp70 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/autotest/client/cros/multimedia/
Dusb_facade_native.py110 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/compiler/tf2tensorrt/segment/
Dsegment.cc92 std::vector<SimpleNode*> out_nodes() const { in out_nodes() function in tensorflow::tensorrt::segment::SimpleNode
261 : [](const SimpleNode* n) { return n->out_nodes(); }; in StableDFS()
646 : node->out_nodes()) { in SegmentGraph()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h138 gtl::iterator_range<NeighborIter> out_nodes() const;
Dgraph.cc207 gtl::iterator_range<NeighborIter> Node::out_nodes() const { in out_nodes() function in tensorflow::Node
/external/tensorflow/tensorflow/compiler/aot/
Dtfcompile.bzl109 # First run tfcompile to generate the list of out_nodes.
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc997 for (auto n : replacee->out_nodes()) in PropagateUpdatedState()
1010 for (auto out : n->out_nodes()) in PropagateUpdatedState()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc759 for (const Node* out : node->out_nodes()) { in ClusterizeNodes()