Function mindspore::dataset::CreateDatasetCache

Function Documentation

inline std::shared_ptr<DatasetCache> mindspore::dataset::CreateDatasetCache(session_id_type id, uint64_t mem_sz, bool spill, std::optional<std::string> hostname = std::nullopt, std::optional<int32_t> port = std::nullopt, std::optional<int32_t> num_connections = std::nullopt, std::optional<int32_t> prefetch_sz = std::nullopt)

Function the create a cache to be attached to a dataset.

Parameters
  • id – A user assigned session id for the current pipeline.

  • mem_sz – Size of the memory set aside for the row caching (default=0 which means unlimited, note that it might bring in the risk of running out of memory on the machine).

  • spill – Spill to disk if out of memory (default=False).

  • hostname – optional host name (default=”127.0.0.1”).

  • port – optional port (default=50052).

  • num_connections – optional number of connections (default=12).

  • prefetch_sz – optional prefetch size (default=20).

Returns

Shared pointer to DatasetCache. If error, nullptr is returned.