Home
last modified time | relevance | path

Searched refs:ZipNode (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dzip_node.cc25 ZipNode::ZipNode(const std::vector<std::shared_ptr<DatasetNode>> &datasets) { in ZipNode() function in mindspore::dataset::ZipNode
30 std::shared_ptr<DatasetNode> ZipNode::Copy() { in Copy()
33 auto node = std::make_shared<ZipNode>(empty_vector); in Copy()
37 void ZipNode::Print(std::ostream &out) const { out << Name(); } in Print()
39 Status ZipNode::ValidateParams() { in ValidateParams()
56 Status ZipNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
65 Status ZipNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate, in GetDatasetSize()
84 Status ZipNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
86 return p->Visit(shared_from_base<ZipNode>(), modified); in Accept()
90 Status ZipNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
[all …]
Dzip_node.h29 class ZipNode : public DatasetNode {
32 explicit ZipNode(const std::vector<std::shared_ptr<DatasetNode>> &datasets);
35 ~ZipNode() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/
Dbindings.cc68 … auto zip = std::make_shared<ZipNode>(std::move(toDatasetNode(self, datasets))); in __anonb78612490102()
292 PYBIND_REGISTER(ZipNode, 2, ([](const py::module *m) { in __anonb78612492602()
293 …(void)py::class_<ZipNode, DatasetNode, std::shared_ptr<ZipNode>>(*m, "ZipNode", "to create a ZipNo… in __anonb78612492602()
295 auto zip = std::make_shared<ZipNode>(datasets); in __anonb78612492602()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h51 class ZipNode; variable
217 virtual Status Visit(std::shared_ptr<ZipNode> node, bool *const modified);
218 virtual Status VisitAfter(std::shared_ptr<ZipNode> node, bool *const modified);
Dpass.cc264 Status IRNodePass::Visit(std::shared_ptr<ZipNode> node, bool *const modified) { in Visit()
267 Status IRNodePass::VisitAfter(std::shared_ptr<ZipNode> node, bool *const modified) { in VisitAfter()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/
Dcache_validation_pass.h72 Status Visit(std::shared_ptr<ZipNode> node, bool *const modified) override;
Dcache_validation_pass.cc97 Status CacheValidationPass::Visit(std::shared_ptr<ZipNode> node, bool *const modified) { in Visit()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dserdes.cc121 RETURN_IF_NOT_OK(ZipNode::from_json(datasets, ds)); in ConstructPipeline()
/third_party/mindspore/tests/ut/cpp/dataset/
Ddeserialize_test.cc433 ds = std::make_shared<ZipNode>(datasets); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc521 auto ds = std::make_shared<ZipNode>(all_datasets); in ZipDataset()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py2937 return cde.ZipNode(children)