Home
last modified time | relevance | path

Searched defs:sym_stride (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorProperties.cpp59 c10::SymInt sym_stride(const Tensor& self, int64_t dim) { in sym_stride() function
H A DTensorShape.cpp1194 …orimpl_meta_symint(const Tensor& self, SymIntArrayRef sym_size, SymIntArrayRef sym_stride, std::op… in as_strided_tensorimpl_meta_symint()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorGeometry.h81 c10::SymInt sym_stride(int64_t dim) const { in sym_stride() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dpromoted_prim_ops.cpp88 void sym_stride(Stack& stack) { in sym_stride() function
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensorBase.h168 c10::SymInt sym_stride(int64_t dim) const { in sym_stride() function
/aosp_15_r20/external/pytorch/torch/masked/maskedtensor/
H A D_ops_refs.py279 def sym_stride(func, *args, **kwargs): function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp304 at::SymIntArrayRef sym_stride, in empty_strided_symint()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyBatchingRegistrations.cpp630 SymIntArrayRef sym_stride, in new_empty_strided_batching_rule()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py6091 def sym_stride(a, dim): function