Home
last modified time | relevance | path

Searched refs:CelebANode (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dceleba_node.cc36 CelebANode::CelebANode(const std::string &dataset_dir, const std::string &usage, in CelebANode() function in mindspore::dataset::CelebANode
46 std::shared_ptr<DatasetNode> CelebANode::Copy() { in Copy()
48 …auto node = std::make_shared<CelebANode>(dataset_dir_, usage_, sampler, decode_, extensions_, cach… in Copy()
52 void CelebANode::Print(std::ostream &out) const { in Print()
56 Status CelebANode::ValidateParams() { in ValidateParams()
68 Status CelebANode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
86 Status CelebANode::GetShardId(int32_t *shard_id) { in GetShardId()
93 Status CelebANode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize()
171 Status CelebANode::to_json(nlohmann::json *out_json) { in to_json()
190 Status CelebANode::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetNode> *ds) { in from_json()
[all …]
Dceleba_node.h31 class CelebANode : public MappableSourceNode {
34CelebANode(const std::string &dataset_dir, const std::string &usage, const std::shared_ptr<Sampler…
38 ~CelebANode() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc59 PYBIND_REGISTER(CelebANode, 2, ([](const py::module *m) { in __anon810239690102()
60 … (void)py::class_<CelebANode, DatasetNode, std::shared_ptr<CelebANode>>(*m, "CelebANode", in __anon810239690102()
64 … auto celebA = std::make_shared<CelebANode>(dataset_dir, usage, toSamplerObj(sampler), decode, in __anon810239690102()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h60 class CelebANode; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dserdes.cc152 RETURN_IF_NOT_OK(CelebANode::from_json(json_obj, ds)); in CreateDatasetNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc882 …auto ds = std::make_shared<CelebANode>(CharToString(dataset_dir), CharToString(usage), sampler_obj… in CelebADataset()
890 …auto ds = std::make_shared<CelebANode>(CharToString(dataset_dir), CharToString(usage), sampler_obj… in CelebADataset()
899 …auto ds = std::make_shared<CelebANode>(CharToString(dataset_dir), CharToString(usage), sampler_obj… in CelebADataset()
/third_party/mindspore/tests/ut/cpp/dataset/
Ddeserialize_test.cc95 …std::shared_ptr<DatasetNode> ds = std::make_shared<CelebANode>(data_dir, usage, sampler, decode, e… in TEST_F()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py5512 … return cde.CelebANode(self.dataset_dir, self.usage, self.sampler, self.decode, self.extensions)