Fix codedex issue for tensor_py

This commit is contained in:
He Wei 2020-09-17 18:28:00 +08:00
parent 63ae2c3639
commit 3ad1acf04c
1 changed files with 4 additions and 1 deletions

View File

@ -137,7 +137,7 @@ class TensorDataNumpy : public TensorData {
ssize_t ndim() const override { return buffer_.ndim; }
/// Data pointer.
void *data() override { return buffer_.ptr; }
void *data() override { return buffer_data(); }
const void *const_data() const override { return buffer_.ptr; }
@ -162,6 +162,9 @@ class TensorDataNumpy : public TensorData {
return py::array(py::dtype(buffer_), buffer_.shape, buffer_.strides, buffer_.ptr, dummyOwner);
}
private:
void *buffer_data() { return buffer_.ptr; }
// The internal buffer.
py::buffer_info buffer_;
};