Home
last modified time | relevance | path

Searched refs:BuildVocabNode (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbuild_vocab_node.cc30 BuildVocabNode::BuildVocabNode(std::shared_ptr<DatasetNode> child, std::shared_ptr<Vocab> vocab, in BuildVocabNode() function in mindspore::dataset::BuildVocabNode
42 std::shared_ptr<DatasetNode> BuildVocabNode::Copy() { in Copy()
44 …std::make_shared<BuildVocabNode>(nullptr, vocab_, columns_, freq_range_, top_k_, special_tokens_, … in Copy()
48 void BuildVocabNode::Print(std::ostream &out) const { in Print()
53 Status BuildVocabNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
63 Status BuildVocabNode::ValidateParams() { in ValidateParams()
92 Status BuildVocabNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
94 return p->Visit(shared_from_base<BuildVocabNode>(), modified); in Accept()
98 Status BuildVocabNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
100 return p->VisitAfter(shared_from_base<BuildVocabNode>(), modified); in AcceptAfter()
Dbuild_vocab_node.h30 class BuildVocabNode : public DatasetNode {
33 BuildVocabNode(std::shared_ptr<DatasetNode> child, std::shared_ptr<Vocab> vocab,
38 ~BuildVocabNode() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/
Dbindings.cc154 PYBIND_REGISTER(BuildVocabNode, 2, ([](const py::module *m) { in __anonb78612490e02()
155 (void)py::class_<BuildVocabNode, DatasetNode, std::shared_ptr<BuildVocabNode>>( in __anonb78612490e02()
160 … std::make_shared<BuildVocabNode>(self, vocab, toStringVector(columns), toIntPair(freq_range), in __anonb78612490e02()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h32 class BuildVocabNode; variable
167 virtual Status Visit(std::shared_ptr<BuildVocabNode> node, bool *const modified);
168 virtual Status VisitAfter(std::shared_ptr<BuildVocabNode> node, bool *const modified);
Dpass.cc138 Status IRNodePass::Visit(std::shared_ptr<BuildVocabNode> node, bool *const modified) { in Visit()
141 Status IRNodePass::VisitAfter(std::shared_ptr<BuildVocabNode> node, bool *const modified) { in VisitAfter()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/
Depoch_ctrl_pass.h55 Status Visit(std::shared_ptr<BuildVocabNode> node, bool *const modified) override;
Depoch_ctrl_pass.cc43 Status EpochCtrlPass::InjectionFinder::Visit(std::shared_ptr<BuildVocabNode> node, bool *const modi… in Visit()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc593 …auto ds = std::make_shared<BuildVocabNode>(IRNode(), vocab, VectorCharToString(columns), freq_rang… in BuildVocabCharIF()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py1163 …vocab_node = cde.BuildVocabNode(ir_tree, vocab, columns, freq_range, top_k, special_tokens, specia…