Home
last modified time | relevance | path

Searched defs:sym_strides (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_meta_info.cpp40 std::vector<std::optional<c10::SymInt>> sym_strides; in build_guard() local
71 auto sym_strides = c10::SymIntArrayRef( in operator ==() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorGeometry.h85 c10::SymIntArrayRef sym_strides() const { in sym_strides() function
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyInterpreter.cpp94 c10::SymIntArrayRef sym_strides(const TensorImpl* self) const override { in sym_strides() function
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensorBase.h245 c10::SymIntArrayRef sym_strides() const { in sym_strides() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_runtime_fusion.cpp562 const IValue& sym_strides = node->ival(attr::striding_inputs_desc); in __anonc321d2750602() local
H A Dtensorexpr_fuser.cpp1382 IValue sym_strides = node->ival(attr::striding_inputs_desc); in createTensorExprOp() local
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp900 c10::SymIntArrayRef ConcretePyInterpreterVTable::sym_strides( in sym_strides() function in torch::detail::__anon56d922760111::ConcretePyInterpreterVTable
/aosp_15_r20/external/pytorch/test/
H A Dtest_dynamic_shapes.py116 sym_strides, argument
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.cpp103 c10::SymIntArrayRef sym_strides(std::vector<SymInt>(v.ndimension(), -1)); in check() local
127 const c10::SymIntArrayRef& sym_strides, in check()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp817 auto sym_strides = in THPVariable_make_wrapper_subclass() local
H A DFunctionsManual.cpp3134 c10::SymIntArrayRef sym_strides, in as_strided_backward()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.h786 c10::SymIntArrayRef sym_strides() const { in sym_strides() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp3205 const auto sym_strides = tensor.sym_strides(); in inferSqueezeGeometry() local