Searched defs:sym_sizes (Results 1 – 14 of 14) sorted by relevance
39 std::vector<std::optional<c10::SymInt>> sym_sizes; in build_guard() local69 auto sym_sizes = c10::SymIntArrayRef( in operator ==() local
78 c10::SymIntArrayRef sym_sizes() const { in sym_sizes() function
49 auto sym_sizes = self_.sym_sizes(); in THPSize_NewFromSymSizes() local
775 c10::SymIntArrayRef ConcretePyInterpreterVTable::sym_sizes( in sym_sizes() function in torch::detail::__anonc6454cf60111::ConcretePyInterpreterVTable
82 c10::SymIntArrayRef sym_sizes(const TensorImpl* self) const override { in sym_sizes() function
242 c10::SymIntArrayRef sym_sizes() const { in sym_sizes() function
759 std::vector<c10::SymInt> sym_sizes; member
1035 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()
113 c10::SymIntArrayRef sym_sizes(size); in to_py_size() local
815 auto sym_sizes = r.symintlist(1); in THPVariable_make_wrapper_subclass() local
577 std::vector<std::optional<c10::SymInt>> sym_sizes; member
126 const c10::SymIntArrayRef& sym_sizes, in check()
614 SymIntArrayRef sym_sizes() const { in sym_sizes() function
3204 const auto sym_sizes = tensor.sym_sizes(); in inferSqueezeGeometry() local