Home
last modified time | relevance | path

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

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