Home
last modified time | relevance | path

Searched full:empty_strided (Results 1 – 25 of 132) sorted by relevance

123456

/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler_tree.py285 aten::empty_strided
297 aten::empty_strided
308 aten::empty_strided
350 aten::empty_strided
355 aten::empty_strided
362 aten::empty_strided
394 aten::empty_strided
409 aten::empty_strided
430 aten::empty_strided
499 aten::empty_strided
[all …]
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_inductor_collectives.py540 .check("buf0 = empty_strided")
572 .check("buf0 = empty_strided")
576 .check("buf5 = empty_strided")
609 .check("buf0 = empty_strided")
610 .check("buf5 = empty_strided")
614 .check("buf6 = empty_strided")
1060 .check("buf0 = empty_strided")
1061 .check("buf6 = empty_strided")
1106 .check("buf0 = empty_strided")
1107 .check("buf6 = empty_strided")
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSort.cpp17 #include <ATen/ops/empty_strided.h>
92 self_ = at::empty_strided(self.sizes(), new_strides_unsort, self.options()); in sort_cuda_kernel()
102 at::empty_strided(self_.sizes(), self_.strides(), self_.options())); in sort_cuda_kernel()
107 at::empty_strided(self_.sizes(), self_.strides(), self_.options().dtype(kLong))); in sort_cuda_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DFactory.cpp49 static Tensor empty_strided( in empty_strided() function
70 TORCH_SELECTIVE_NAME("aten::empty_strided"), in TORCH_LIBRARY_IMPL()
71 TORCH_FN(at::native::vulkan::ops::empty_strided)); in TORCH_LIBRARY_IMPL()
H A DClone.cpp8 #include <ATen/ops/empty_strided.h>
30 self = at::empty_strided(src.sizes(), src.strides(), src.options()); in clone()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/
H A DMetalAten.mm90 static Tensor empty_strided(
108 m.impl(TORCH_SELECTIVE_NAME("aten::empty_strided"), TORCH_FN(empty_strided));
/aosp_15_r20/external/executorch/extension/tensor/
H A Dtensor_ptr_maker.cpp86 empty_strided(std::move(sizes), std::move(strides), type, dynamism); in random_strided()
99 TensorPtr empty_strided( in empty_strided() function
123 empty_strided(std::move(sizes), std::move(strides), type, dynamism); in full_strided()
H A Dtensor_ptr_maker.h297 TensorPtr empty_strided(
326 return empty_strided(
349 return empty_strided(std::move(sizes), {}, type, dynamism);
/aosp_15_r20/external/pytorch/test/
H A Dtest_dynamic_shapes.py793 r = torch.empty_strided((a0, 7), (1, a0), device="meta")
804 self.assertTrue(cf(torch.empty_strided((u0, 2), (2, 1), device="meta")))
805 self.assertTrue(cf(torch.empty_strided((2, u0), (1, 2), device="meta")))
809 self.assertTrue(cf(torch.empty_strided((u0,), (1,), device="meta")))
810 self.assertTrue(cf(torch.empty_strided((1,), (u0,), device="meta")))
817 torch.empty_strided(
846 self.assertEqual(cf(torch.empty_strided((u0, 2), (2, 1), device="meta")), 0)
847 self.assertEqual(cf(torch.empty_strided((2, u0), (1, 2), device="meta")), 0)
848 self.assertEqual(cf(torch.empty_strided((u0,), (1,), device="meta")), 0)
849 self.assertEqual(cf(torch.empty_strided((1,), (u0,), device="meta")), 0)
[all …]
H A Dtest_tensor_creation_ops.py748 # Note: This test failed on XLA since its test cases are created by empty_strided which
786 torch.empty_strided((4, 3, 2), (10, 3, 1), device=device).fill_(1.0),
787 torch.empty_strided((4, 3, 2), (10, 0, 3), device=device).fill_(1.0),
788 torch.empty_strided((4, 3, 2), (10, 1, 2), device=device).fill_(1.0),
789 torch.empty_strided((4, 3, 2), (4, 2, 1), device=device)[:, :, ::2].fill_(1.0),
790 torch.empty_strided((4, 3, 2), (10, 1, 1), device=device).fill_(1.0),
791 torch.empty_strided((4, 1, 1, 2), (10, 0, 0, 2), device=device).fill_(1.0),
792 torch.empty_strided((4, 2, 3), (10, 3, 3), device=device).fill_(1.0))
2516 # allows them then empty_strided can as well.
2518 empty_strided = torch.empty_strided(shape, strides, device=device)
[all …]
/aosp_15_r20/external/executorch/docs/source/
H A Dextension-tensor.md294 And `empty_strided()` creates an uninitialized tensor with sizes and strides specified.
297 auto tensor = empty_strided({2, 3}, {3, 1});
383 | `at::empty_strided(sizes, strides)` | `empty_strided(sizes, strides)` |
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dmeta_utils.py783 return torch.empty_strided(
1183 lambda: torch.empty_strided(
1191 r.real_tensor = torch.empty_strided(
1279 lambda: torch.empty_strided(
1288 … r.real_tensor = torch.empty_strided( # type: ignore[attr-defined]
1451 lambda: torch.empty_strided(
1462 r.real_tensor = torch.empty_strided(
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/test/
H A Dtensor_factory_test.cpp312 Tensor expected = at::empty_strided( in TEST_F()
349 Tensor expected = at::empty_strided( in TEST_F()
386 Tensor expected = at::empty_strided( in TEST_F()
425 Tensor expected = at::empty_strided( in TEST_F()
475 Tensor expected = at::empty_strided( in TEST_F()
506 Tensor expected = at::empty_strided( in TEST_F()
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py196 "empty_strided",
263 return torch.empty_strided(shape, strides, dtype=dtype, device=device)
696 return torch.empty_strided(
2284 return torch.empty_strided(
2295 out = torch.empty_strided(
2585 empty_strided = _make_prim( variable
2586 …schema="empty_strided(SymInt[] shape, SymInt[] strides, *, ScalarType dtype, Device device, bool r…
2589 impl_aten=torch.empty_strided,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml181 - empty_strided
217 - empty_strided
H A DLinearAlgebraUtils.h24 #include <ATen/ops/empty_strided.h>
103 auto copy = at::empty_strided(copy_sizes, copy_strides, src.options());
222 a_buffer = at::empty_strided(a.sizes(), a.strides(), a.options()) in batch_iterator_with_broadcasting()
421 auto strided_to = at::empty_strided(original_tensor.sizes(), in same_stride_to()
H A DWeightNorm.cpp16 #include <ATen/ops/empty_strided.h>
59 auto norm = at::empty_strided(g.sizes(), g.strides(), g.options().dtype(dtype)); in weight_norm_cpu()
/aosp_15_r20/external/pytorch/test/typing/pass/
H A Dcreation_ops.py86 # torch.empty/empty_like/empty_strided
91 torch.empty_strided((2, 3), (1, 2))
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dflex_decoding.py12 from ..lowering import empty, empty_strided, lowerings
407 buf_M = empty_strided(
413 buf_L = empty_strided(
/aosp_15_r20/external/pytorch/test/typing/reveal/
H A Dtensor_constructors.py90 # torch.empty/empty_like/empty_strided
93 reveal_type(torch.empty_strided((2, 3), (1, 2))) # E: {Tensor}
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_ck_backend.py194 a = torch.empty_strided((50257, 32768), (1, 50304), **tensor_options)
195 b = torch.empty_strided((32768, 768), (768, 1), **tensor_options)
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h82 virtual at::Tensor empty_strided( in empty_strided() function
89 return at::empty_strided( in empty_strided()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/testing_util/
H A Dtensor_factory.h279 t = empty_strided(sizes, strides);
319 t = empty_strided(sizes, strides);
526 at::Tensor empty_strided(
532 return at::empty_strided(
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect200 aten::empty_strided
201 aten::empty_strided.out
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_common.cpp35 #include <ATen/ops/empty_strided.h>
388 new_tensor_handle(at::empty_strided(sizes, strides, options)); in aoti_torch_empty_strided()
416 : at::empty_strided(sizes, strides, options)); in aoti_torch_create_tensor_from_blob()

123456