Home
last modified time | relevance | path

Searched refs:split_node (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dsplit_cat.py99 def _get_split_args_default(split_node): argument
104 if split_node.op == "call_method":
107 get_arg_value(split_node, 0, input_kwarg),
108 get_arg_value(split_node, 1, split_size_kwarg),
109 get_arg_value(split_node, 2, dim_kwarg) or default_dim_value,
165 split_node = match.nodes[0]
167 split_input, split_size, split_dim = _get_split_args(split_node)
171 if not is_node_meta_valid(split_node):
172 log.debug("example value absent for node: %s", split_node)
174 assert isinstance(split_node.meta["example_value"], (list, tuple))
[all …]
H A Dpost_grad.py612 split_node, cat_node = split_nodes[0], cat_nodes[0]
614 if get_arg_value(split_node, 2, "dim") != get_arg_value(cat_node, 1, "dim"):
620 split_sizes = get_arg_value(split_node, 1, "split_sizes")
925 split_node, cat_node = split_nodes[0], cat_nodes[0]
932 dim = get_arg_value(split_node, 2, "dim")
937 split_sizes = get_arg_value(split_node, 1, "split_sizes")
H A Dddp_fusion.py331 split_node = call_function(
339 with graph.inserting_after(split_node):
343 graph, operator.getitem, (split_node, idx)
/aosp_15_r20/external/executorch/backends/arm/_passes/
H A Dconvert_split_to_slice.py34 split_node = node
35 input_node = split_node.all_input_nodes[0]
36 output_nodes = split_node.users.copy()
39 split_lengths = split_node.args[1]
40 dim = split_node.args[2] if len(split_node.args) > 2 else 0
60 with graph_module.graph.inserting_before(split_node):
68 slice_node.meta = split_node.meta.copy()
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DSplitLayerSubTensorMutator.cpp83 … auto *split_node = arm_compute::utils::cast::polymorphic_downcast<SplitLayerNode *>(node); in mutate() local
85 const int axis = split_node->axis(); in mutate()
86 const unsigned int num_splits = split_node->num_splits(); in mutate()
95 …std::tie(std::ignore, coords) = split_node->compute_output_descriptor(input_tensor->desc(), num_sp… in mutate()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_alias_analysis.py30 split_node = graph.findNode("aten::split")
33 alias_db.may_contain_alias(next(split_node.inputs()), split_node.output())
37 alias_db.may_contain_alias(next(split_node.inputs()), next(graph.inputs()))
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfixup_onnx_controlflow.cpp146 Node* split_node = in ConvertSequenceDependencies() local
148 loop_output->replaceAllUsesWith(split_node->output()); in ConvertSequenceDependencies()
149 split_node->i_(attr::keepdims, 0); in ConvertSequenceDependencies()
150 split_node->addInput(loop_output); in ConvertSequenceDependencies()
151 split_node->insertAfter(loop_node); in ConvertSequenceDependencies()
152 split_node->output()->setType(loop_output->type()); in ConvertSequenceDependencies()
153 split_node->copyMetadata(loop_node); in ConvertSequenceDependencies()
162 new_outputs.push_back(split_node->output()); in ConvertSequenceDependencies()
H A Dpeephole.cpp973 auto split_node = it->input()->node(); in removeSequenceSplitConcat() local
977 split_node->hasAttribute(attr::axis) ? split_node->i(attr::axis) : 0; in removeSequenceSplitConcat()
978 const auto split_keepdims = split_node->hasAttribute(attr::keepdims) in removeSequenceSplitConcat()
979 ? split_node->i(attr::keepdims) in removeSequenceSplitConcat()
985 const bool has_input_split = split_node->inputs().size() == 2; in removeSequenceSplitConcat()
999 concat_node->output()->replaceAllUsesWith(split_node->input()); in removeSequenceSplitConcat()
/aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/alloc/src/collections/
H A Dlinked_list.rs340 split_node: Option<NonNull<Node<T>>>, in split_off_before_node()
347 if let Some(mut split_node) = split_node { in split_off_before_node()
351 first_part_tail = split_node.as_mut().prev.take(); in split_off_before_node()
371 self.head = Some(split_node); in split_off_before_node()
383 split_node: Option<NonNull<Node<T>>>, in split_off_after_node()
391 if let Some(mut split_node) = split_node { in split_off_after_node()
395 second_part_head = split_node.as_mut().next.take(); in split_off_after_node()
415 self.tail = Some(split_node); in split_off_after_node()
966 let split_node = if at - 1 <= len - 1 - (at - 1) { in split_off() localVariable
983 unsafe { self.split_off_after_node(split_node, at) } in split_off()
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/alloc/src/collections/
H A Dlinked_list.rs340 split_node: Option<NonNull<Node<T>>>, in split_off_before_node()
347 if let Some(mut split_node) = split_node { in split_off_before_node()
351 first_part_tail = split_node.as_mut().prev.take(); in split_off_before_node()
371 self.head = Some(split_node); in split_off_before_node()
383 split_node: Option<NonNull<Node<T>>>, in split_off_after_node()
391 if let Some(mut split_node) = split_node { in split_off_after_node()
395 second_part_head = split_node.as_mut().next.take(); in split_off_after_node()
415 self.tail = Some(split_node); in split_off_after_node()
966 let split_node = if at - 1 <= len - 1 - (at - 1) { in split_off() localVariable
983 unsafe { self.split_off_after_node(split_node, at) } in split_off()
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/alloc/src/collections/
H A Dlinked_list.rs340 split_node: Option<NonNull<Node<T>>>, in split_off_before_node()
347 if let Some(mut split_node) = split_node { in split_off_before_node()
351 first_part_tail = split_node.as_mut().prev.take(); in split_off_before_node()
371 self.head = Some(split_node); in split_off_before_node()
383 split_node: Option<NonNull<Node<T>>>, in split_off_after_node()
391 if let Some(mut split_node) = split_node { in split_off_after_node()
395 second_part_head = split_node.as_mut().next.take(); in split_off_after_node()
415 self.tail = Some(split_node); in split_off_after_node()
966 let split_node = if at - 1 <= len - 1 - (at - 1) { in split_off() localVariable
983 unsafe { self.split_off_after_node(split_node, at) } in split_off()
/aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/alloc/src/collections/
H A Dlinked_list.rs340 split_node: Option<NonNull<Node<T>>>, in split_off_before_node()
347 if let Some(mut split_node) = split_node { in split_off_before_node()
351 first_part_tail = split_node.as_mut().prev.take(); in split_off_before_node()
371 self.head = Some(split_node); in split_off_before_node()
383 split_node: Option<NonNull<Node<T>>>, in split_off_after_node()
391 if let Some(mut split_node) = split_node { in split_off_after_node()
395 second_part_head = split_node.as_mut().next.take(); in split_off_after_node()
415 self.tail = Some(split_node); in split_off_after_node()
966 let split_node = if at - 1 <= len - 1 - (at - 1) { in split_off() localVariable
983 unsafe { self.split_off_after_node(split_node, at) } in split_off()
/aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/alloc/src/collections/
H A Dlinked_list.rs340 split_node: Option<NonNull<Node<T>>>, in split_off_before_node()
347 if let Some(mut split_node) = split_node { in split_off_before_node()
351 first_part_tail = split_node.as_mut().prev.take(); in split_off_before_node()
371 self.head = Some(split_node); in split_off_before_node()
383 split_node: Option<NonNull<Node<T>>>, in split_off_after_node()
391 if let Some(mut split_node) = split_node { in split_off_after_node()
395 second_part_head = split_node.as_mut().next.take(); in split_off_after_node()
415 self.tail = Some(split_node); in split_off_after_node()
966 let split_node = if at - 1 <= len - 1 - (at - 1) { in split_off() localVariable
983 unsafe { self.split_off_after_node(split_node, at) } in split_off()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc827 Node* split_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
886 .Finalize(graph, &split_node)); in CreateConcatAndSplitNodesForInputTensor()
889 output_from_concat_node = split_node; in CreateConcatAndSplitNodesForInputTensor()
911 graph->AddEdge(split_node, output_index, edge->dst(), edge->dst_input()); in CreateConcatAndSplitNodesForInputTensor()
1011 Node* split_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local
1019 .Finalize(graph, &split_node)); in CreateConcatAndSplitNodesForOutputTensor()
1025 if (edge->dst() == split_node) { in CreateConcatAndSplitNodesForOutputTensor()
1052 graph->AddEdge(split_node, output_index, output_edge->dst(), in CreateConcatAndSplitNodesForOutputTensor()
1517 TF_ASSIGN_OR_RETURN(Node * split_node, init_graph->AddNode(split_def)); in InitializeShardedVarOnTPU()
1518 split_node->set_assigned_device_name(cpu_device); in InitializeShardedVarOnTPU()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc669 Node* split_node; member
680 Node* control_predecessor, Node* split_node, in CreatePadNode() argument
686 graph->NewName(absl::StrCat(split_node->name(), "/paddings"))); in CreatePadNode()
689 paddings_def.set_device(split_node->assigned_device_name()); in CreatePadNode()
708 absl::StrCat(split_node->name(), "/pad_shard_", split_index))); in CreatePadNode()
710 pad_def.set_device(split_node->assigned_device_name()); in CreatePadNode()
713 pad_def.add_input(absl::StrCat(split_node->name(), ":", split_index)); in CreatePadNode()
716 pad_node->set_assigned_device_name(split_node->assigned_device_name()); in CreatePadNode()
718 graph->AddEdge(split_node, split_index, pad_node, 0); in CreatePadNode()
767 TF_ASSIGN_OR_RETURN(Node * split_node, graph->AddNode(split_def)); in CreateSplitNode()
[all …]
/aosp_15_r20/external/icing/icing/legacy/index/
H A Dicing-dynamic-trie.cc1650 Node *split_node = storage_->GetMutableNode(best_node_index); in Insert() local
1651 const char *prev_suffix = storage_->GetSuffix(split_node->next_index()); in Insert()
1689 split_node->set_next_index(storage_->GetNextArrayIndex(split_next)); in Insert()
1690 split_node->set_is_leaf(false); in Insert()
1691 split_node->set_log2_num_children(0); in Insert()
1696 split_node = child_node; in Insert()
1701 split_node->set_next_index(storage_->GetNextArrayIndex(split_next)); in Insert()
1702 split_node->set_is_leaf(false); in Insert()
1703 split_node->set_log2_num_children(1); in Insert()
/aosp_15_r20/external/openscreen/third_party/abseil/src/absl/container/internal/
H A Dbtree.h2331 node_type *split_node;
2333 split_node = new_leaf_node(parent);
2334 node->split(insert_position, split_node, mutable_allocator());
2335 if (rightmost_ == node) rightmost_ = split_node;
2337 split_node = new_internal_node(parent);
2338 node->split(insert_position, split_node, mutable_allocator());
2343 node = split_node;
/aosp_15_r20/external/e2fsprogs/debugfs/
H A Dextent_cmds.ct53 split_node, split;
/aosp_15_r20/external/private-join-and-compute/third_party/abseil-cpp-20230125.2/absl/container/internal/
H A Dbtree.h2673 node_type *split_node;
2675 split_node = new_leaf_node(parent);
2676 node->split(insert_position, split_node, mutable_allocator());
2677 if (rightmost() == node) mutable_rightmost() = split_node;
2679 split_node = new_internal_node(parent);
2680 node->split(insert_position, split_node, mutable_allocator());
2685 node = split_node;
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/container/internal/
Dbtree.h2673 node_type *split_node;
2675 split_node = new_leaf_node(parent);
2676 node->split(insert_position, split_node, mutable_allocator());
2677 if (rightmost() == node) mutable_rightmost() = split_node;
2679 split_node = new_internal_node(parent);
2680 node->split(insert_position, split_node, mutable_allocator());
2685 node = split_node;
/aosp_15_r20/external/webrtc/third_party/abseil-cpp/absl/container/internal/
H A Dbtree.h2630 node_type *split_node;
2632 split_node = new_leaf_node(parent);
2633 node->split(insert_position, split_node, mutable_allocator());
2634 if (rightmost() == node) mutable_rightmost() = split_node;
2636 split_node = new_internal_node(parent);
2637 node->split(insert_position, split_node, mutable_allocator());
2642 node = split_node;
/aosp_15_r20/external/abseil-cpp/absl/container/internal/
H A Dbtree.h2699 node_type *split_node;
2701 split_node = new_leaf_node(node->position() + 1, parent);
2702 node->split(insert_position, split_node, mutable_allocator());
2703 if (rightmost() == node) mutable_rightmost() = split_node;
2705 split_node = new_internal_node(node->position() + 1, parent);
2706 node->split(insert_position, split_node, mutable_allocator());
2711 node = split_node;
/aosp_15_r20/external/cronet/third_party/abseil-cpp/absl/container/internal/
H A Dbtree.h2699 node_type *split_node;
2701 split_node = new_leaf_node(node->position() + 1, parent);
2702 node->split(insert_position, split_node, mutable_allocator());
2703 if (rightmost() == node) mutable_rightmost() = split_node;
2705 split_node = new_internal_node(node->position() + 1, parent);
2706 node->split(insert_position, split_node, mutable_allocator());
2711 node = split_node;
/aosp_15_r20/external/angle/third_party/abseil-cpp/absl/container/internal/
H A Dbtree.h2699 node_type *split_node;
2701 split_node = new_leaf_node(node->position() + 1, parent);
2702 node->split(insert_position, split_node, mutable_allocator());
2703 if (rightmost() == node) mutable_rightmost() = split_node;
2705 split_node = new_internal_node(node->position() + 1, parent);
2706 node->split(insert_position, split_node, mutable_allocator());
2711 node = split_node;
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc1796 NodeDef* split_node) { in HoistChainForSplit() argument
1798 const string& split_name = split_node->name(); in HoistChainForSplit()
1809 const int value_slot = split_node->op() == "SplitV" ? 0 : 1; in HoistChainForSplit()
1810 const string orig_input = split_node->input(value_slot); in HoistChainForSplit()
1811 split_node->set_input(value_slot, cur_copy->name()); in HoistChainForSplit()
1812 ctx().node_map->UpdateInput(split_node->name(), orig_input, in HoistChainForSplit()
1817 while (cur_tail != split_node) { in HoistChainForSplit()

12