Searched refs:ConcatNode (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | concat_node.cc | 34 ConcatNode::ConcatNode(const std::vector<std::shared_ptr<DatasetNode>> &datasets, in ConcatNode() function in mindspore::dataset::ConcatNode
45 std::shared_ptr<DatasetNode> ConcatNode::Copy() { in Copy()
48 auto node = std::make_shared<ConcatNode>(std::vector<std::shared_ptr<DatasetNode>>(), sampler, in Copy()
53 void ConcatNode::Print(std::ostream &out) const { out << Name(); } in Print()
55 Status ConcatNode::ValidateParams() { in ValidateParams()
81 Status ConcatNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize()
127 Status ConcatNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
144 Status ConcatNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
146 return p->Visit(shared_from_base<ConcatNode>(), modified); in Accept()
150 Status ConcatNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
[all …]
|
| D | concat_node.h | 30 class ConcatNode : public DatasetNode {
33 explicit ConcatNode(const std::vector<std::shared_ptr<DatasetNode>> &datasets,
39 ~ConcatNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 167 PYBIND_REGISTER(ConcatNode, 2, ([](const py::module *m) { in __anonb78612491002()
168 … (void)py::class_<ConcatNode, DatasetNode, std::shared_ptr<ConcatNode>>(*m, "ConcatNode", in __anonb78612491002()
172 auto concat = std::make_shared<ConcatNode>(datasets, toSamplerObj(sampler), in __anonb78612491002()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 38 class ConcatNode; variable
169 virtual Status Visit(std::shared_ptr<ConcatNode> node, bool *const modified);
170 virtual Status VisitAfter(std::shared_ptr<ConcatNode> node, bool *const modified);
|
| D | pass.cc | 144 Status IRNodePass::Visit(std::shared_ptr<ConcatNode> node, bool *const modified) { in Visit()
147 Status IRNodePass::VisitAfter(std::shared_ptr<ConcatNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | cache_validation_pass.h | 48 Status Visit(std::shared_ptr<ConcatNode> node, bool *const modified) override;
|
| D | cache_validation_pass.cc | 49 Status CacheValidationPass::Visit(std::shared_ptr<ConcatNode> node, bool *const modified) { in Visit()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | serdes.cc | 124 RETURN_IF_NOT_OK(ConcatNode::from_json(json_obj, datasets, ds)); in ConstructPipeline()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | deserialize_test.cc | 527 …ds = std::make_shared<ConcatNode>(datasets, sampler, children_flag_and_nums, children_start_end_in… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 445 auto ds = std::make_shared<ConcatNode>(all_datasets); in ConcatDataset()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 2992 …return cde.ConcatNode(children, self._sampler, self._children_flag_and_nums, self._children_start_…
|