Browse Source

Fix codedex issue for tensor_py

tags/v1.0.0
He Wei 5 years ago
parent
commit
3ad1acf04c
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindspore/ccsrc/pybind_api/ir/tensor_py.cc

+ 4
- 1
mindspore/ccsrc/pybind_api/ir/tensor_py.cc View File

@@ -137,7 +137,7 @@ class TensorDataNumpy : public TensorData {
ssize_t ndim() const override { return buffer_.ndim; } ssize_t ndim() const override { return buffer_.ndim; }


/// Data pointer. /// Data pointer.
void *data() override { return buffer_.ptr; }
void *data() override { return buffer_data(); }


const void *const_data() const override { return buffer_.ptr; } 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); return py::array(py::dtype(buffer_), buffer_.shape, buffer_.strides, buffer_.ptr, dummyOwner);
} }


private:
void *buffer_data() { return buffer_.ptr; }

// The internal buffer. // The internal buffer.
py::buffer_info buffer_; py::buffer_info buffer_;
}; };


Loading…
Cancel
Save