| 170 | PyObject* TensorSize_New(Py_ssize_t len) { return TensorSize_Type.tp_alloc(&TensorSize_Type, len); } |
| 171 | |
| 172 | PyObject* TensorSize_NewFromShape(const Shape& size) { |
| 173 | PyObjectPtr self(TensorSize_New(size.NumAxes())); |
| 174 | if (self.get()) { |
| 175 | for (int i = 0; i < size.NumAxes(); ++i) { |
| 176 | PyTuple_SET_ITEM(self.get(), i, PyLong_FromLongLong(size.At(i))); |
| 177 | } |
| 178 | } |
| 179 | return self.release(); |
| 180 | } |
| 181 | |
| 182 | Shape TensorSize_AsShape(PyObject* self) { |
| 183 | if (!TensorSize_Check(self)) { |
no test coverage detected