Searched refs:TakeNode (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | take_node.cc | 32 TakeNode::TakeNode(std::shared_ptr<DatasetNode> child, int32_t count) : take_count_(count) { this->… in TakeNode() function in mindspore::dataset::TakeNode
34 std::shared_ptr<DatasetNode> TakeNode::Copy() { in Copy()
35 auto node = std::make_shared<TakeNode>(nullptr, take_count_); in Copy()
39 void TakeNode::Print(std::ostream &out) const { out << (Name() + "(num_rows:" + std::to_string(take… in Print()
42 Status TakeNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
51 Status TakeNode::ValidateParams() { in ValidateParams()
63 Status TakeNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimat… in GetDatasetSize()
77 Status TakeNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
79 return p->Visit(shared_from_base<TakeNode>(), modified); in Accept()
83 Status TakeNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
[all …]
|
| D | take_node.h | 29 class TakeNode : public DatasetNode {
32 explicit TakeNode(std::shared_ptr<DatasetNode> child, int32_t count);
35 ~TakeNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 269 PYBIND_REGISTER(TakeNode, 2, ([](const py::module *m) { in __anonb78612492202()
270 (void)py::class_<TakeNode, DatasetNode, std::shared_ptr<TakeNode>>(*m, "TakeNode", in __anonb78612492202()
273 auto take = std::make_shared<TakeNode>(self, count); in __anonb78612492202()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 48 class TakeNode; variable
211 virtual Status Visit(std::shared_ptr<TakeNode> node, bool *const modified);
212 virtual Status VisitAfter(std::shared_ptr<TakeNode> node, bool *const modified);
|
| D | pass.cc | 246 Status IRNodePass::Visit(std::shared_ptr<TakeNode> node, bool *const modified) { in Visit()
249 Status IRNodePass::VisitAfter(std::shared_ptr<TakeNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | node_removal_pass.h | 61 Status Visit(std::shared_ptr<TakeNode> node, bool *const modified) override;
|
| D | node_removal_pass.cc | 46 Status NodeRemovalPass::RemovalNodes::Visit(std::shared_ptr<TakeNode> node, bool *const modified) { in Visit()
|
| D | cache_validation_pass.h | 66 Status Visit(std::shared_ptr<TakeNode> node, bool *const modified) override;
|
| D | cache_validation_pass.cc | 85 Status CacheValidationPass::Visit(std::shared_ptr<TakeNode> node, bool *const modified) { in Visit()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | tree_modifying_function_test.cc | 287 std::shared_ptr<TakeNode> ds6_to_insert = std::make_shared<TakeNode>(nullptr, 12); in TEST_F()
|
| D | deserialize_test.cc | 223 ds = std::make_shared<TakeNode>(ds, 6); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | serdes.cc | 202 RETURN_IF_NOT_OK(TakeNode::from_json(json_obj, ds, result)); in CreateDatasetOperationNode()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 510 auto ds = std::make_shared<TakeNode>(input->IRNode(), count); in TakeDataset()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 2919 return cde.TakeNode(children[0], self.count)
|