Home
last modified time | relevance | path

Searched defs:sym_sizes (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/
H A Dkernel_meta_info.cpp39 std::vector<std::optional<c10::SymInt>> sym_sizes; in build_guard() local
69 auto sym_sizes = c10::SymIntArrayRef( in operator ==() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorGeometry.h78 c10::SymIntArrayRef sym_sizes() const { in sym_sizes() function
/aosp_15_r20/external/pytorch/torch/csrc/
H A DSize.cpp49 auto sym_sizes = self_.sym_sizes(); in THPSize_NewFromSymSizes() local
H A DPyInterpreter.cpp775 c10::SymIntArrayRef ConcretePyInterpreterVTable::sym_sizes( in sym_sizes() function in torch::detail::__anon56d922760111::ConcretePyInterpreterVTable
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPyInterpreter.cpp82 c10::SymIntArrayRef sym_sizes(const TensorImpl* self) const override { in sym_sizes() function
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensorBase.h242 c10::SymIntArrayRef sym_sizes() const { in sym_sizes() function
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction.h759 std::vector<c10::SymInt> sym_sizes; member
H A DFunctionsManual.cpp1035 Tensor unsqueeze_to(const Tensor& self, c10::SymIntArrayRef sym_sizes) { in unsqueeze_to()
1050 c10::SymIntArrayRef sym_sizes) { in unsqueeze_to()
1066 c10::SymIntArrayRef sym_sizes) { in unsqueeze_to()
2175 c10::SymIntArrayRef sym_sizes, in split_backward()
3133 c10::SymIntArrayRef sym_sizes, in as_strided_backward()
H A Dpython_function.cpp113 c10::SymIntArrayRef sym_sizes(size); in to_py_size() local
H A Dpython_variable.cpp815 auto sym_sizes = r.symintlist(1); in THPVariable_make_wrapper_subclass() local
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dcompiled_autograd.h577 std::vector<std::optional<c10::SymInt>> sym_sizes; member
H A Dguards.cpp126 const c10::SymIntArrayRef& sym_sizes, in check()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.h614 SymIntArrayRef sym_sizes() const { in sym_sizes() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp3204 const auto sym_sizes = tensor.sym_sizes(); in inferSqueezeGeometry() local