Searched refs:CacheLookupNode (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | cache_lookup_node.cc | 29 CacheLookupNode::CacheLookupNode(std::shared_ptr<DatasetNode> child, std::shared_ptr<SamplerObj> sa… in CacheLookupNode() function in mindspore::dataset::CacheLookupNode
35 void CacheLookupNode::Print(std::ostream &out) const { out << Name(); } in Print()
37 std::shared_ptr<DatasetNode> CacheLookupNode::Copy() { in Copy()
39 auto node = std::make_shared<CacheLookupNode>(nullptr, sampler, cache_); in Copy()
44 Status CacheLookupNode::ValidateParams() { in ValidateParams()
49 Status CacheLookupNode::Build(std::vector<std::shared_ptr<DatasetOp>> *node_ops) { in Build()
61 Status CacheLookupNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
63 return p->Visit(shared_from_base<CacheLookupNode>(), modified); in Accept()
67 Status CacheLookupNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
69 return p->VisitAfter(shared_from_base<CacheLookupNode>(), modified); in AcceptAfter()
[all …]
|
| D | cache_lookup_node.h | 28 class CacheLookupNode : public DatasetNode, public SamplerObj {
31 CacheLookupNode(std::shared_ptr<DatasetNode> child, std::shared_ptr<SamplerObj> sampler,
35 ~CacheLookupNode() = default;
86 std::shared_ptr<CacheLookupNode> lookup_node_copy_;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 34 class CacheLookupNode; variable
174 virtual Status Visit(std::shared_ptr<CacheLookupNode> node, bool *const modified);
175 virtual Status VisitAfter(std::shared_ptr<CacheLookupNode> node, bool *const modified);
|
| D | pass.cc | 151 Status IRNodePass::Visit(std::shared_ptr<CacheLookupNode> node, bool *const modified) { in Visit()
154 Status IRNodePass::VisitAfter(std::shared_ptr<CacheLookupNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/ |
| D | repeat_pass.h | 95 Status VisitAfter(std::shared_ptr<CacheLookupNode> node, bool *const modified) override;
|
| D | repeat_pass.cc | 190 Status RepeatPass::VisitAfter(std::shared_ptr<CacheLookupNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | cache_transform_pass.cc | 225 …auto cache_lookup_node = std::make_shared<CacheLookupNode>(nullptr, leaf_sampler, cached_node->Get… in InjectMappableCacheNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | minddata_node.cc | 187 auto cache_lookup_sampler = std::dynamic_pointer_cast<CacheLookupNode>(sampler_); in Build()
|