Browse Source

update liteapi with include

tags/v1.3.0
Xiao Tianci 5 years ago
parent
commit
9c7e509a5a
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