| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Sort.cpp | 17 #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()
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| D | Factory.cpp | 49 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()
|
| D | Clone.cpp | 8 #include <ATen/ops/empty_strided.h>
30 self = at::empty_strided(src.sizes(), src.strides(), src.options()); in clone()
|
| /external/pytorch/test/profiler/ |
| D | test_profiler_tree.py | 285 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 …]
|
| /external/pytorch/aten/src/ATen/native/metal/ |
| D | MetalAten.mm | 90 static Tensor empty_strided(
108 m.impl(TORCH_SELECTIVE_NAME("aten::empty_strided"), TORCH_FN(empty_strided));
|
| /external/executorch/extension/tensor/ |
| D | tensor_ptr_maker.cpp | 86 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()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ts_native_functions.yaml | 181 - empty_strided
217 - empty_strided
|
| D | WeightNorm.cpp | 16 #include <ATen/ops/empty_strided.h>
59 auto norm = at::empty_strided(g.sizes(), g.strides(), g.options().dtype(dtype)); in weight_norm_cpu()
|
| D | LinearAlgebraUtils.h | 24 #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()
|
| /external/pytorch/test/distributed/ |
| D | test_inductor_collectives.py | 540 .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")
|
| /external/pytorch/test/typing/pass/ |
| D | creation_ops.py | 86 # torch.empty/empty_like/empty_strided
91 torch.empty_strided((2, 3), (1, 2))
|
| /external/pytorch/docs/source/ |
| D | deterministic.rst | 26 * :func:`torch.empty_strided`
|
| D | jit_unsupported.rst | 49 * :func:`torch.empty_strided`
|
| /external/pytorch/test/typing/reveal/ |
| D | tensor_constructors.py | 90 # torch.empty/empty_like/empty_strided
93 reveal_type(torch.empty_strided((2, 3), (1, 2))) # E: {Tensor}
|
| /external/executorch/docs/source/ |
| D | extension-tensor.md | 294 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)` |
|
| /external/pytorch/torch/_lazy/ |
| D | tensor_factory_functions.py | 20 torch.empty_strided,
|
| /external/pytorch/test/inductor/ |
| D | test_ck_backend.py | 194 a = torch.empty_strided((50257, 32768), (1, 50304), **tensor_options)
195 b = torch.empty_strided((32768, 768), (768, 1), **tensor_options)
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | codegen.h | 82 virtual at::Tensor empty_strided( in empty_strided() function
89 return at::empty_strided( in empty_strided()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | extension_backend_test.cpp | 52 m.impl("aten::empty_strided", empty_strided_override); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/_inductor/kernel/ |
| D | flex_decoding.py | 12 from ..lowering import empty, empty_strided, lowerings
407 buf_M = empty_strided(
413 buf_L = empty_strided(
|
| /external/pytorch/torch/utils/ |
| D | _device.py | 17 torch.empty_strided,
|
| /external/pytorch/test/ |
| D | test_dynamic_shapes.py | 793 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 …]
|
| /external/pytorch/torch/_subclasses/ |
| D | meta_utils.py | 783 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(
|
| /external/pytorch/torch/utils/jit/ |
| D | log_extract.py | 40 return torch.empty_strided(size=size, stride=stride, device=device, dtype=dtype)
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_aten_core_operators.expect | 200 aten::empty_strided
201 aten::empty_strided.out
|