| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/ |
| D | cache_client.cc | 26 CacheClient::Builder::Builder() in Builder()
35 Status CacheClient::Builder::Build(std::shared_ptr<CacheClient> *out) { in Build()
38 …*out = std::make_shared<CacheClient>(session_id_, cache_mem_sz_, spill_, hostname_, port_, num_con… in Build()
43 Status CacheClient::Builder::SanityCheck() { in SanityCheck()
57 CacheClient::CacheClient(session_id_type session_id, uint64_t cache_mem_sz, bool spill, std::string… in CacheClient() function in mindspore::dataset::CacheClient
71 CacheClient::~CacheClient() { in ~CacheClient()
98 void CacheClient::Print(std::ostream &out) const { in Print()
106 std::string CacheClient::GetHostname() const { return comm_->GetHostname(); } in GetHostname()
107 int32_t CacheClient::GetPort() const { return comm_->GetPort(); } in GetPort()
109 Status CacheClient::WriteRow(const TensorRow &row, row_id_type *row_id_from_server) const { in WriteRow()
[all …]
|
| D | cache_client.h | 49 class CacheClient {
131 Status Build(std::shared_ptr<CacheClient> *out);
147 …CacheClient(session_id_type session_id, uint64_t cache_mem_sz, bool spill, std::string hostname, i…
151 ~CacheClient();
207 friend std::ostream &operator<<(std::ostream &out, const CacheClient &cc) {
316 Status Init(CacheClient *cc);
351 CacheClient *cc_;
|
| D | cache_request.h | 38 class CacheClient; variable
98 friend class CacheClient; variable
186 friend class CacheClient; variable
187 explicit CacheRowRequest(const CacheClient *cc);
193 Status SerializeCacheRowRequest(const CacheClient *cc, const TensorRow &row);
234 BatchFetchRequest(const CacheClient *cc, const std::vector<row_id_type> &row_id);
253 explicit CreateCacheRequest(CacheClient *cc, const CacheClientInfo &cinfo, uint64_t cache_mem_sz,
264 CacheClient *cc_;
495 explicit BatchCacheRowsRequest(const CacheClient *cc, int64_t addr, int32_t num_ele);
|
| D | cache_request.cc | 38 Status CacheRowRequest::SerializeCacheRowRequest(const CacheClient *cc, const TensorRow &row) { in SerializeCacheRowRequest()
129 CacheRowRequest::CacheRowRequest(const CacheClient *cc) in CacheRowRequest()
140 BatchFetchRequest::BatchFetchRequest(const CacheClient *cc, const std::vector<row_id_type> &row_id) in BatchFetchRequest()
208 CreateCacheRequest::CreateCacheRequest(CacheClient *cc, const CacheClientInfo &cinfo, uint64_t cach… in CreateCacheRequest()
348 BatchCacheRowsRequest::BatchCacheRowsRequest(const CacheClient *cc, int64_t addr, int32_t num_ele) in BatchCacheRowsRequest()
|
| D | cache_grpc_client.h | 59 friend class CacheClient; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/cache/ |
| D | bindings.cc | 24 PYBIND_REGISTER(CacheClient, 0, ([](const py::module *m) { in __anon19b8a73b0102()
25 (void)py::class_<CacheClient, std::shared_ptr<CacheClient>>(*m, "CacheClient") in __anon19b8a73b0102()
29 std::shared_ptr<CacheClient> cc; in __anon19b8a73b0102()
30 CacheClient::Builder builder; in __anon19b8a73b0102()
39 .def("GetStat", [](CacheClient &cc) { in __anon19b8a73b0102()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | cache_base_op.h | 47 …CacheBase(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> cache_clien…
78 std::shared_ptr<CacheClient> GetCacheClient() { return cache_client_; } in GetCacheClient()
80 …void SetCacheClient(std::shared_ptr<CacheClient> cache_client) { cache_client_ = std::move(cache_c… in SetCacheClient()
88 std::shared_ptr<CacheClient> cache_client_;
|
| D | cache_merge_op.h | 59 Status AsyncSendCacheRequest(const std::shared_ptr<CacheClient> &cc, const TensorRow &row);
81 std::shared_ptr<CacheClient> cache_client);
128 std::shared_ptr<CacheClient> cache_client_;
|
| D | cache_op.h | 43 CacheOp(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> cache_client,
|
| D | cache_lookup_op.h | 36 …CacheLookupOp(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> cache_c… in CacheLookupOp()
|
| D | cache_merge_op.cc | 43 std::shared_ptr<CacheClient> cache_client) in CacheMergeOp()
253 Status CacheMergeOp::TensorRowCacheRequest::AsyncSendCacheRequest(const std::shared_ptr<CacheClient… in AsyncSendCacheRequest()
|
| D | cache_op.cc | 32 CacheOp::CacheOp(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> cache… in CacheOp()
|
| D | cache_base_op.cc | 46 CacheBase::CacheBase(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> c… in CacheBase()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | cache_op_test.cc | 34 using mindspore::dataset::CacheClient;
66 CacheClient::Builder builder; in TEST_F()
73 std::shared_ptr<CacheClient> myClient; in TEST_F()
158 CacheClient::Builder builder; in TEST_F()
161 std::shared_ptr<CacheClient> myClient; in TEST_F()
266 CacheClient::Builder builder; in TEST_F()
268 std::shared_ptr<CacheClient> myClient; in TEST_F()
383 CacheClient::Builder builder; in TEST_F()
385 std::shared_ptr<CacheClient> myClient; in TEST_F()
453 CacheClient::Builder ccbuilder; in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/perf/ |
| D | cache_perf_run.h | 80 CacheClient::Builder cache_builder_;
91 std::shared_ptr<CacheClient> cc_;
|
| D | cache_pipeline_run.h | 97 CacheClient::Builder cache_builder_;
108 std::shared_ptr<CacheClient> cc_;
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | cache_client.py | 19 from mindspore._c_dataengine import CacheClient
78 …self.cache_client = CacheClient(session_id, size, spilling, hostname, port, num_connections, prefe…
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/cache/ |
| D | pre_built_dataset_cache.h | 34 explicit PreBuiltDatasetCache(std::shared_ptr<CacheClient> cc) in PreBuiltDatasetCache()
|
| D | dataset_cache_impl.h | 81 std::shared_ptr<CacheClient> cache_client_;
|
| D | dataset_cache_impl.cc | 31 CacheClient::Builder builder; in Build()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/ |
| D | cache_transform_pass.h | 32 class CacheClient; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/ |
| D | pybind_conversion.h | 76 std::shared_ptr<DatasetCache> toDatasetCache(std::shared_ptr<CacheClient> cc);
|
| D | pybind_conversion.cc | 187 std::shared_ptr<DatasetCache> toDatasetCache(std::shared_ptr<CacheClient> cc) { in toDatasetCache()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/ |
| D | bindings.cc | 62 [](std::shared_ptr<DatasetNode> self, std::shared_ptr<CacheClient> cc) { in __anonb78612490102()
|