| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/ |
| D | transfer_node.cc | 34 TransferNode::TransferNode(std::shared_ptr<DatasetNode> child, std::string queue_name, std::string … in TransferNode() function in mindspore::dataset::TransferNode
46 std::shared_ptr<DatasetNode> TransferNode::Copy() { in Copy()
47 …auto node = std::make_shared<TransferNode>(nullptr, queue_name_, device_type_, device_id_, send_ep… in Copy()
52 void TransferNode::Print(std::ostream &out) const { in Print()
58 Status TransferNode::ValidateParams() { in ValidateParams()
69 Status TransferNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
107 Status TransferNode::Accept(IRNodePass *const p, bool *const modified) { in Accept()
109 return p->Visit(shared_from_base<TransferNode>(), modified); in Accept()
113 Status TransferNode::AcceptAfter(IRNodePass *const p, bool *const modified) { in AcceptAfter()
115 return p->VisitAfter(shared_from_base<TransferNode>(), modified); in AcceptAfter()
[all …]
|
| D | transfer_node.h | 29 class TransferNode : public DatasetNode {
32 …TransferNode(std::shared_ptr<DatasetNode> child, std::string queue_name, std::string device_type, …
36 ~TransferNode() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 279 PYBIND_REGISTER(TransferNode, 2, ([](const py::module *m) { in __anonb78612492402()
280 … (void)py::class_<TransferNode, DatasetNode, std::shared_ptr<TransferNode>>(*m, "TransferNode", in __anonb78612492402()
285 auto transfer = std::make_shared<TransferNode>( in __anonb78612492402()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/ |
| D | pass.h | 50 class TransferNode; variable
215 virtual Status Visit(std::shared_ptr<TransferNode> node, bool *const modified);
216 virtual Status VisitAfter(std::shared_ptr<TransferNode> node, bool *const modified);
|
| D | pass.cc | 258 Status IRNodePass::Visit(std::shared_ptr<TransferNode> node, bool *const modified) { in Visit()
261 Status IRNodePass::VisitAfter(std::shared_ptr<TransferNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | epoch_ctrl_pass.h | 69 Status VisitAfter(std::shared_ptr<TransferNode> node, bool *const modified) override;
|
| D | epoch_ctrl_pass.cc | 60 Status EpochCtrlPass::InjectionFinder::VisitAfter(std::shared_ptr<TransferNode> node, bool *const m… in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/ |
| D | repeat_pass.h | 102 Status VisitAfter(std::shared_ptr<TransferNode> node, bool *const modified) override;
|
| D | repeat_pass.cc | 209 Status RepeatPass::VisitAfter(std::shared_ptr<TransferNode> node, bool *const modified) { in VisitAfter()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 70 friend class TransferNode; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ |
| D | serdes.cc | 200 RETURN_IF_NOT_OK(TransferNode::from_json(json_obj, ds, result)); in CreateDatasetOperationNode()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | deserialize_test.cc | 469 ds = std::make_shared<TransferNode>(ds, "queue", "type", 1, true, 10, true); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 207 …std::make_shared<TransferNode>(shared_from_this()->IRNode(), CharToString(queue_name), CharToStrin… in DeviceQueueCharIF()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 90 friend class TransferNode; variable
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 3180 …return cde.TransferNode(children[0], self.queue_name, self.device_type, self.device_id, self._send…
|