| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_from_dataset.py | 59 def test_config(freq_range, top_k): argument
107 def test_config(texts, top_k, special_tokens, special_first): argument
126 def test_config(columns, freq_range, top_k, s): argument
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | arg_max_fusion.cc | 21 void ArgMaxFusion::Init(const bool keep_dims, const bool out_max_value, const int64_t top_k, const … in Init()
32 void ArgMaxFusion::set_top_k(const int64_t top_k) { (void)this->AddAttr(kTopK, MakeValue(top_k)); } in set_top_k()
|
| D | arg_min_fusion.cc | 21 void ArgMinFusion::Init(bool keep_dims, bool out_max_value, int64_t top_k, int64_t axis) { in Init()
30 void ArgMinFusion::set_top_k(int64_t top_k) { (void)this->AddAttr(kTopK, MakeValue(top_k)); } in set_top_k()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | build_vocab_node.cc | 32 … int64_t top_k, const std::vector<std::string> &special_tokens, bool special_first) in BuildVocabNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | build_vocab_op.cc | 31 … std::pair<int64_t, int64_t> freq_r, int64_t top_k, const std::vector<std::string> &tokens, in BuildVocabOp() argument
|
| /third_party/mindspore/mindspore/dataset/text/ |
| D | utils.py | 41 …def from_dataset(cls, dataset, columns=None, freq_range=None, top_k=None, special_tokens=None, spe… argument
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 158 … py::tuple freq_range, int64_t top_k, py::list special_tokens, bool special_first) { in __anonb78612490e02()
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 488 optional uint32 top_k = 1 [default = 1]; field
504 optional uint32 top_k = 2 [default = 1]; field
1572 optional int32 top_k = 2; field
1596 optional int32 top_k = 14; field
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 589 … const std::pair<int64_t, int64_t> &freq_range, int64_t top_k, in BuildVocabCharIF() argument
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 241 PrimitivePtr MindIR_ArgMaxFusion_CreatePrimitive(int64_t axis, int64_t top_k, bool keep_dims, bool … in MindIR_ArgMaxFusion_CreatePrimitive()
296 void MindIR_ArgMaxFusion_SetTopK(PrimitivePtr *primitive, int64_t top_k) { in MindIR_ArgMaxFusion_SetTopK()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 1108 def build_vocab(self, columns, freq_range, top_k, special_tokens, special_first): argument
|