Browse Source

!15050 Fix compiling bug of Typecast in MindSpore lite

From: @tiancixiao
Reviewed-by: @heleiwang,@liucunwei,@liucunwei
Signed-off-by: @liucunwei,@liucunwei
pull/15050/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
292ac61312
1 changed files with 2 additions and 4 deletions
  1. +2
    -4
      mindspore/ccsrc/minddata/dataset/liteapi/include/transforms.h

+ 2
- 4
mindspore/ccsrc/minddata/dataset/liteapi/include/transforms.h View File

@@ -205,10 +205,8 @@ class RandomChoice final : public TensorTransform {
class TypeCast final : public TensorTransform {
public:
/// \brief Constructor.
/// \param[in] data_type mindspore.dtype to be cast to.
explicit TypeCast(std::string data_type) : TypeCast(StringToChar(data_type)) {}

explicit TypeCast(const std::vector<char> &data_type);
/// \param[in] data_type mindspore::DataType to be cast to.
explicit TypeCast(mindspore::DataType data_type);

/// \brief Destructor
~TypeCast() = default;


Loading…
Cancel
Save