| 37 | { |
| 38 | public: |
| 39 | static TensorBatch Create(int capacity) |
| 40 | { |
| 41 | PyObject *tensorBatch = capi().TensorBatch_Create(capacity); |
| 42 | CheckCAPIError(); |
| 43 | NVCV_ASSERT(tensorBatch != nullptr); |
| 44 | py::object pytensorBatch = py::reinterpret_steal<py::object>(tensorBatch); |
| 45 | |
| 46 | return TensorBatch(pytensorBatch); |
| 47 | } |
| 48 | |
| 49 | void pushBack(Tensor tensor) |
| 50 | { |
nothing calls this directly
no test coverage detected