提交 6af4e7c2 编写于 作者: E Eric

Addressing comments

上级 a4ef5b26
...@@ -23,7 +23,7 @@ namespace mindspore { ...@@ -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
......
...@@ -30,10 +30,11 @@ class TensorOp; ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册