Home
last modified time | relevance | path

Searched refs:split_with_sizes (Results 1 – 25 of 51) sorted by relevance

123

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_pattern_matcher.py837 split_with_sizes = torch.ops.aten.split_with_sizes.default(a, [8, 24], 1)
838 getitem = split_with_sizes[0]
839 getitem_1 = split_with_sizes[1]
850 split_with_sizes = torch.ops.aten.split_with_sizes.default(a, [8, 8, 16], 1)
851 getitem = split_with_sizes[0]
852 getitem_1 = split_with_sizes[1]
853 getitem_2 = split_with_sizes[2]
864 split_with_sizes = torch.ops.aten.split_with_sizes.default(
867 cat = torch.ops.aten.cat.default(split_with_sizes, 0)
877 x = torch.ops.aten.split_with_sizes.default(a, [3, 2, 3], dim=1)
[all …]
H A Dtest_cpu_repro.py2827 split_with_sizes = torch.ops.aten.split_with_sizes.default(
2830 getitem = split_with_sizes[0]
2831 getitem_1 = split_with_sizes[1]
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp285 tensor.split_with_sizes(/*split_sizes=*/chunk_sizes, /*dim=*/dim); in _broadcast_out_impl()
332 ? tensor.split_with_sizes(/*split_sizes=*/*chunk_sizes, /*dim=*/dim) in _broadcast_out_impl()
384 out_tensor.split_with_sizes(/*split_sizes=*/chunk_sizes, /*dim=*/dim); in _broadcast_out_impl()
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpost_grad.py607 split_nodes = filter_nodes(match.nodes, aten.split_with_sizes)
902 aten.split_with_sizes,
922 split_nodes = filter_nodes(match.nodes, aten.split_with_sizes)
957 aten.split_with_sizes,
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dinference_mode.cpp482 b = s_view.split_with_sizes({1, 1}); in TEST()
485 c = s.split_with_sizes({1, 1}); in TEST()
/aosp_15_r20/external/executorch/backends/arm/quantizer/quantization_annotation/
H A Dgeneric_annotator.py49 torch.ops.aten.split_with_sizes.default,
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dpreprocess_for_onnx.cpp96 case aten::split_with_sizes: in FuseWithListUnpack()
/aosp_15_r20/external/executorch/backends/arm/test/ops/
H A Dtest_split.py38 return x.split_with_sizes(split_sizes=split_sizes, dim=dim)
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dnative_ops.cpp736 at::native::split_with_sizes(self, split_sizes.vec(), dim); in __anon75e5f0514602()
745 aten::split_with_sizes,
760 at::native::split_with_sizes(self, split_sizes.vec(), dim); in __anon75e5f0514902()
/aosp_15_r20/external/pytorch/docs/source/
H A Dtensor_view.rst81 - :meth:`~torch.Tensor.split_with_sizes`
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupMPI.cpp810 srcFlatData.split_with_sizes(c10::IntArrayRef(send_lengthsL), 0); in alltoall()
828 dstFlatData.split_with_sizes(c10::IntArrayRef(recv_lengthsL), 0); in alltoall()
/aosp_15_r20/external/executorch/backends/qualcomm/_passes/
H A Dlayout_transform.py72 exir_ops.edge.aten.split_with_sizes.default,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cpp196 at::split_with_sizes(metadata, {offsets.numel(), nt_sizes.numel()}, 0); in NestedTensor_to_padded_tensor_cuda()
/aosp_15_r20/external/pytorch/torch/distributed/_tools/
H A Druntime_estimator.py56 aten.split_with_sizes,
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api202 split_with_sizes
H A Dannotated_ops253 split_with_sizes, view/reshape
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_special_ops.cpp250 auto result = at::split_with_sizes( in __anonedd36e380202()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset13.py119 def split_with_sizes(g: jit_utils.GraphContext, self, split_sizes, dim, _outputs=None): function
134 return split_with_sizes(g, self, split_sizes, dim, _outputs)
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect483 aten::split_with_sizes
/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_tensor_ops.py730 aten.split_with_sizes.default,
/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py1343 a_splits = a.split_with_sizes(split_sizes, dim=-1)
1344 b_splits = b.split_with_sizes(split_sizes, dim=-1)
1345 c_splits = c.split_with_sizes(split_sizes, dim=-1)
1348 nt_splits = nt.split_with_sizes(split_sizes, dim=-1)
1369 lambda: torch.split_with_sizes(nt, split_sizes, dim=1),
1376 lambda: torch.split_with_sizes(nt, split_sizes, dim=0),
1384 lambda: torch.split_with_sizes(nt_noncontiguous, split_sizes, dim=-1),
1392 lambda: torch.split_with_sizes(nt, bad_split_sizes, dim=-1),
3312 splits = nt.split_with_sizes([2, 3], dim=-1)
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/timm_train/
H A Dbotnet26t_256_training.txt220 Operator: aten.split_with_sizes.default
H A Dlevit_128_training.txt281 Operator: aten.split_with_sizes.default
H A Deca_botnext26ts_256_training.txt276 Operator: aten.split_with_sizes.default
H A Deca_halonext26ts_training.txt322 Operator: aten.split_with_sizes.default

123