Home
last modified time | relevance | path

Searched defs:empty_strided (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DFactory.cpp49 static Tensor empty_strided( in empty_strided() function
/aosp_15_r20/external/executorch/extension/tensor/
H A Dtensor_ptr_maker.cpp99 TensorPtr empty_strided( in empty_strided() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dcodegen.h82 virtual at::Tensor empty_strided( in empty_strided() function
H A Dcuda_codegen.cpp1250 at::Tensor CudaCodeGen::empty_strided( in empty_strided() function in torch::jit::tensorexpr::CudaCodeGen
H A Dllvm_codegen.cpp446 at::Tensor LLVMCodeGen::empty_strided( in empty_strided() function in LLVMCodeGen
/aosp_15_r20/external/pytorch/torch/_prims/
H A D__init__.py2585 empty_strided = _make_prim( variable
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py5354 def empty_strided( function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2865 def empty_strided( function