Searched refs:MnistNode (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | mnist_node.cc | 33 MnistNode::MnistNode(std::string dataset_dir, std::string usage, std::shared_ptr<SamplerObj> sample… in MnistNode() function in mindspore::dataset::MnistNode
37 std::shared_ptr<DatasetNode> MnistNode::Copy() { in Copy()
39 auto node = std::make_shared<MnistNode>(dataset_dir_, usage_, sampler, cache_); in Copy()
43 void MnistNode::Print(std::ostream &out) const { out << Name(); } in Print()
45 Status MnistNode::ValidateParams() { in ValidateParams()
56 Status MnistNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
76 Status MnistNode::GetShardId(int32_t *const shard_id) { in GetShardId()
83 Status MnistNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize()
102 Status MnistNode::to_json(nlohmann::json *out_json) { in to_json()
119 Status MnistNode::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetNode> *ds) { in from_json()
[all …]
|
| D | mnist_node.h | 29 class MnistNode : public MappableSourceNode {
32 MnistNode(std::string dataset_dir, std::string usage, std::shared_ptr<SamplerObj> sampler,
36 ~MnistNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/ |
| D | bindings.cc | 242 PYBIND_REGISTER(MnistNode, 2, ([](const py::module *m) { in __anon810239691d02()
243 … (void)py::class_<MnistNode, DatasetNode, std::shared_ptr<MnistNode>>(*m, "MnistNode", in __anon810239691d02()
246 … auto mnist = std::make_shared<MnistNode>(dataset_dir, usage, toSamplerObj(sampler), nullptr); in __anon810239691d02()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 66 class MnistNode; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | serdes.cc | 170 RETURN_IF_NOT_OK(MnistNode::from_json(json_obj, ds)); in CreateDatasetNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 1236 …auto ds = std::make_shared<MnistNode>(CharToString(dataset_dir), CharToString(usage), sampler_obj,… in MnistDataset()
1242 …auto ds = std::make_shared<MnistNode>(CharToString(dataset_dir), CharToString(usage), sampler_obj,… in MnistDataset()
1248 …auto ds = std::make_shared<MnistNode>(CharToString(dataset_dir), CharToString(usage), sampler_obj,… in MnistDataset()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | deserialize_test.cc | 59 std::shared_ptr<DatasetNode> ds = std::make_shared<MnistNode>(data_dir, usage, sampler, nullptr); in TEST_F()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 3511 return cde.MnistNode(self.dataset_dir, self.usage, self.sampler)
|