Searched refs:py_seq (Results 1 – 1 of 1) sorted by relevance
108 static std::vector<int64_t> seq_to_aten_shape(PyObject* py_seq) { in seq_to_aten_shape() argument109 int ndim = PySequence_Length(py_seq); in seq_to_aten_shape()115 auto item = THPObjectPtr(PySequence_GetItem(py_seq, i)); in seq_to_aten_shape()