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

!15050 Fix compiling bug of Typecast in MindSpore lite 

From: @tiancixiao
Reviewed-by: @heleiwang,@liucunwei,@liucunwei
Signed-off-by: @liucunwei,@liucunwei
This commit is contained in:
mindspore-ci-bot 2021-04-13 15:22:12 +08:00 committed by Gitee
parent 789c28d4d8 9c7e509a5a
commit 292ac61312
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/minddata/dataset/liteapi/include/transforms.h
View File

@ -205,10 +205,8 @@ class RandomChoice final : public TensorTransform {
class TypeCast final : public TensorTransform { class TypeCast final : public TensorTransform {
public: public:
/// \brief Constructor. /// \brief Constructor.
/// \param[in] data_type mindspore.dtype to be cast to. /// \param[in] data_type mindspore::DataType to be cast to.
explicit TypeCast(std::string data_type) : TypeCast(StringToChar(data_type)) {} explicit TypeCast(mindspore::DataType data_type);
explicit TypeCast(const std::vector<char> &data_type);
/// \brief Destructor /// \brief Destructor
~TypeCast() = default; ~TypeCast() = default;