Home
last modified time | relevance | path

Searched defs:out_node (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_rewrite.cc82 for (const auto& out_node : out_nodes) { in ReplaceReduce() local
94 for (const auto& out_node : out_nodes) { in ReplaceReduce() local
150 for (const auto& out_node : out_nodes) { in ReplaceBroadcast() local
191 for (const auto& out_node : out_nodes_map[send_dev]) { in ReplaceBroadcast() local
234 for (const auto& out_node : out_nodes_map[recv_dev]) { in ReplaceBroadcast() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dsubgraph.cc220 Node** out_node) { in AddNode()
237 Node** out_node) { in AddNode()
257 Node** out_node) { in AddNode()
276 Node** out_node) { in AddNode()
/aosp_15_r20/frameworks/base/tools/aapt2/format/proto/
H A DProtoSerialize.cpp778 static void SerializeXmlCommon(const xml::Node& node, pb::XmlNode* out_node) { in SerializeXmlCommon()
784 void SerializeXmlToPb(const xml::Element& el, pb::XmlNode* out_node, in SerializeXmlToPb()
835 void SerializeXmlResourceToPb(const xml::XmlResource& resource, pb::XmlNode* out_node, in SerializeXmlResourceToPb()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc388 for (const auto out_node : switch_node->out_nodes()) { in AddNoOpAfterLastIteration() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dgraph_opt.cpp416 auto out_node = out->node(); in replaceListOutputWithTuple() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_execution_state.cc319 Node** out_node) override { in AddNode()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp700 auto out_node = in clamp_node_creator() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dselect_algorithm.py722 def make_kernel_render(out_node): argument
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/alloc/src/collections/btree/
H A Dmap.rs228 let mut out_node = match root.borrow_mut().force() { in clone() localVariable
251 let mut out_node = out_root.push_internal_level(alloc.clone()); in clone() localVariable
/aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/alloc/src/collections/btree/
H A Dmap.rs228 let mut out_node = match root.borrow_mut().force() { in clone() localVariable
251 let mut out_node = out_root.push_internal_level(alloc.clone()); in clone() localVariable
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/alloc/src/collections/btree/
H A Dmap.rs228 let mut out_node = match root.borrow_mut().force() { in clone() localVariable
251 let mut out_node = out_root.push_internal_level(alloc.clone()); in clone() localVariable
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/alloc/src/collections/btree/
H A Dmap.rs228 let mut out_node = match root.borrow_mut().force() { in clone() localVariable
251 let mut out_node = out_root.push_internal_level(alloc.clone()); in clone() localVariable
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/alloc/src/collections/btree/
H A Dmap.rs228 let mut out_node = match root.borrow_mut().force() { in clone() localVariable
251 let mut out_node = out_root.push_internal_level(alloc.clone()); in clone() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc1192 Node* out_node = out_edge->dst(); in SelectGraphToExecute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmodel.cc2130 std::shared_ptr<Node>* out_node) { in AddNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc3929 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local