p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Addressing comments

This commit is contained in:
Eric 2020-08-04 14:00:40 -04:00
parent a4ef5b26a1
commit 6af4e7c226
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/minddata/dataset/api/execute.cc
View File

@ -23,7 +23,7 @@ namespace mindspore {
namespace dataset {
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) {
// Build the op

3
mindspore/ccsrc/minddata/dataset/include/execute.h
View File

@ -30,10 +30,11 @@ class TensorOp;
namespace api {
// class to run tensor operations in eager mode
class Execute {
public:
/// \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
/// \param[inout] input - the tensor to be transformed