Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dcifar10_node.cc34 Cifar10Node::Cifar10Node(const std::string &dataset_dir, const std::string &usage, std::shared_ptr<… in Cifar10Node() function in mindspore::dataset::Cifar10Node
38 std::shared_ptr<DatasetNode> Cifar10Node::Copy() { in Copy()
40 auto node = std::make_shared<Cifar10Node>(dataset_dir_, usage_, sampler, cache_); in Copy()
44 void Cifar10Node::Print(std::ostream &out) const { in Print()
48 Status Cifar10Node::ValidateParams() { in ValidateParams()
60 Status Cifar10Node::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
80 Status Cifar10Node::GetShardId(int32_t *shard_id) { in GetShardId()
87 Status Cifar10Node::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool esti… in GetDatasetSize()
109 Status Cifar10Node::to_json(nlohmann::json *out_json) { in to_json()
126 Status Cifar10Node::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetNode> *ds) { in from_json()
[all …]
Dcifar10_node.h29 class Cifar10Node : public MappableSourceNode {
32Cifar10Node(const std::string &dataset_dir, const std::string &usage, std::shared_ptr<SamplerObj> …
36 ~Cifar10Node() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc71 PYBIND_REGISTER(Cifar10Node, 2, ([](const py::module *m) { in __anon810239690302()
72 … (void)py::class_<Cifar10Node, DatasetNode, std::shared_ptr<Cifar10Node>>(*m, "Cifar10Node", in __anon810239690302()
75 … auto cifar10 = std::make_shared<Cifar10Node>(dataset_dir, usage, toSamplerObj(sampler), nullptr); in __anon810239690302()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h62 class Cifar10Node; variable
/third_party/mindspore/tests/ut/cpp/dataset/
Ddeserialize_test.cc154 std::shared_ptr<DatasetNode> ds = std::make_shared<Cifar10Node>(data_dir, usage, sampler, cache); in TEST_F()
503 …std::shared_ptr<DatasetNode> ds = std::make_shared<Cifar10Node>(data_dir, usage, sampler, some_cac… in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dserdes.cc154 RETURN_IF_NOT_OK(Cifar10Node::from_json(json_obj, ds)); in CreateDatasetNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc907 …auto ds = std::make_shared<Cifar10Node>(CharToString(dataset_dir), CharToString(usage), sampler_ob… in Cifar10Dataset()
913 …auto ds = std::make_shared<Cifar10Node>(CharToString(dataset_dir), CharToString(usage), sampler_ob… in Cifar10Dataset()
920 …auto ds = std::make_shared<Cifar10Node>(CharToString(dataset_dir), CharToString(usage), sampler_ob… in Cifar10Dataset()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py4564 return cde.Cifar10Node(self.dataset_dir, self.usage, self.sampler)