Browse Source

!6425 Fix codedex issue for tensor_py

Merge pull request !6425 from hewei/fix_codedex_0917
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
a79943ca98
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

@@ -138,7 +138,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; }

@@ -163,6 +163,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_;
};


Loading…
Cancel
Save