Browse Source

Addressing comments

tags/v0.7.0-beta
Eric 5 years ago
parent
commit
6af4e7c226
2 changed files with 3 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/minddata/dataset/api/execute.cc
  2. +2
    -1
      mindspore/ccsrc/minddata/dataset/include/execute.h

+ 1
- 1
mindspore/ccsrc/minddata/dataset/api/execute.cc View File

@@ -23,7 +23,7 @@ namespace mindspore {
namespace dataset { namespace dataset {
namespace api { namespace api {


Execute::Execute(const std::shared_ptr<TensorOperation> &op) : op_(std::move(op)) {}
Execute::Execute(std::shared_ptr<TensorOperation> op) : op_(std::move(op)) {}


std::shared_ptr<tensor::MSTensor> Execute::operator()(std::shared_ptr<tensor::MSTensor> input) { std::shared_ptr<tensor::MSTensor> Execute::operator()(std::shared_ptr<tensor::MSTensor> input) {
// Build the op // Build the op


+ 2
- 1
mindspore/ccsrc/minddata/dataset/include/execute.h View File

@@ -30,10 +30,11 @@ class TensorOp;


namespace api { namespace api {


// class to run tensor operations in eager mode
class Execute { class Execute {
public: public:
/// \brief Constructor /// \brief Constructor
explicit Execute(const std::shared_ptr<TensorOperation> &op);
explicit Execute(std::shared_ptr<TensorOperation> op);


/// \brief callable function to execute the TensorOperation in eager mode /// \brief callable function to execute the TensorOperation in eager mode
/// \param[inout] input - the tensor to be transformed /// \param[inout] input - the tensor to be transformed


Loading…
Cancel
Save