Home
last modified time | relevance | path

Searched defs:layout_opt (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DEmptyTensor.cpp266 std::optional<Layout> layout_opt, in empty_cpu()
301 std::optional<Layout> layout_opt, in empty_strided_cpu()
357 std::optional<Layout> layout_opt, in empty_meta()
377 std::optional<Layout> layout_opt, in empty_symint_meta()
411 std::optional<Layout> layout_opt, in empty_strided_meta()
446 std::optional<Layout> layout_opt, in empty_strided_symint_meta()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorFactories.cu54 …cuda(IntArrayRef size, std::optional<ScalarType> dtype_opt, std::optional<Layout> layout_opt, std:… in empty_cuda()
80 …e, IntArrayRef stride, std::optional<ScalarType> dtype_opt, std::optional<Layout> layout_opt, std:… in empty_strided_cuda()
278 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in tril_indices_cuda()
354 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in triu_indices_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMetaTensor.cpp17 std::optional<Layout> layout_opt, in empty_meta_symint()
35 std::optional<Layout> layout_opt, in empty_strided_meta_symint()
H A DTensorFactories.cpp247 … empty_cpu(IntArrayRef size, std::optional<ScalarType> dtype_opt, std::optional<Layout> layout_opt, in empty_cpu()
281 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in empty_permuted_symint()
320 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in empty_strided_cpu()
504 std::optional<Layout> layout_opt, in new_empty_symint()
1181 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in tril_indices_cpu()
1231 …std::optional<Layout> layout_opt, std::optional<Device> device_opt, std::optional<bool> pin_memory… in triu_indices_cpu()
H A DConstraints.cpp82 std::optional<Layout> layout_opt, in _make_dep_token_cpu()
H A DTensorFactories.h66 int64_t row, int64_t col, std::optional<Layout> layout_opt) { in check_args()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DEmptyTensor.cpp26 std::optional<Layout> layout_opt, in empty_cuda()
67 std::optional<Layout> layout_opt, in empty_strided_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DTensorFactory.cpp78 std::optional<Layout> layout_opt, in empty_mps()
90 std::optional<Layout> layout_opt, in empty_strided_mps()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/csrc/
H A DOpenRegMem.cpp64 std::optional<c10::Layout> layout_opt, in empty_openreg()
83 std::optional<c10::Layout> layout_opt, in empty_strided_openreg()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DEmptyTensor.cpp24 std::optional<Layout> layout_opt, in empty_mps()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h86 std::optional<c10::Layout> layout_opt, in empty_strided()
H A Dcuda_codegen.cpp1254 std::optional<c10::Layout> layout_opt, in empty_strided()
H A Dllvm_codegen.cpp450 std::optional<c10::Layout> layout_opt, in empty_strided()
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp137 …ArrayRef stride, std::optional<at::ScalarType> dtype_opt, std::optional<at::Layout> layout_opt, st… in custom_empty_strided()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py306 layout_opt=None, argument
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorOptions.h326 std::optional<Layout> layout_opt() const noexcept { in layout_opt() function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp387 std::optional<at::Layout> layout_opt, in custom_empty_strided()