Home
last modified time | relevance | path

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

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