Home
last modified time | relevance | path

Searched refs:dataset_type (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/schema/
Dbindings.cc54 …set_dataset_type", [](SchemaObj &self, std::string dataset_type) { self.set_dataset_type(dataset_t… in __anon6ef38fe40102() argument
/third_party/mindspore/tests/ut/cpp/dataset/
Dceleba_op_test.cc40 const std::string &dataset_type = "all") { in Celeba() argument
54 …elebAOp> so = std::make_shared<CelebAOp>(num_workers, dir, op_connector_size, decode, dataset_type, in Celeba()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/
Ddatasets.h164 …bool Save(std::string dataset_path, int32_t num_files = 1, std::string dataset_type = "mindrecord"…
165 return SaveCharIF(StringToChar(dataset_path), num_files, StringToChar(dataset_type));
308 …F(const std::vector<char> &dataset_path, int32_t num_files, const std::vector<char> &dataset_type);
366 void set_dataset_type(std::string dataset_type);
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/
Dtree_consumer.h103 …explicit SaveToDisk(std::string dataset_path, int32_t num_files = 1, std::string dataset_type = "m…
104 … TreeConsumer(), dataset_path_(dataset_path), num_files_(num_files), dataset_type_(dataset_type) {} in TreeConsumer()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc236 const std::vector<char> &dataset_type) { in SaveCharIF() argument
248 … = std::make_unique<SaveToDisk>(CharToString(dataset_path), num_files, CharToString(dataset_type)); in SaveCharIF()
731 void SchemaObj::set_dataset_type(std::string dataset_type) { data_->dataset_type_ = dataset_type.da… in set_dataset_type() argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Ddatasets.h184 …bool Save(std::string dataset_path, int32_t num_files = 1, std::string dataset_type = "mindrecord"…
185 return SaveCharIF(StringToChar(dataset_path), num_files, StringToChar(dataset_type));
468 …F(const std::vector<char> &dataset_path, int32_t num_files, const std::vector<char> &dataset_type);
540 void set_dataset_type(std::string dataset_type);