Program Listing for File config.h
↰ Return to documentation for file (include/config.h
)
#ifndef MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_CONFIG_H
#define MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_CONFIG_H
#include <cstdint>
#include <string>
#include <vector>
#include "include/api/dual_abi_helper.h"
namespace mindspore {
namespace dataset {
// Config operations for setting and getting the configuration.
namespace config {
bool set_seed(int32_t seed);
uint32_t get_seed();
bool set_prefetch_size(int32_t prefetch_size);
int32_t get_prefetch_size();
bool set_num_parallel_workers(int32_t num_parallel_workers);
int32_t get_num_parallel_workers();
bool set_monitor_sampling_interval(int32_t interval);
int32_t get_monitor_sampling_interval();
bool set_callback_timeout(int32_t timeout);
int32_t get_callback_timeout();
bool load(const std::vector<char> &file);
inline bool load(std::string file) { return load(StringToChar(file)); }
} // namespace config
} // namespace dataset
} // namespace mindspore
#endif // MINDSPORE_CCSRC_MINDDATA_DATASET_INCLUDE_DATASET_CONFIG_H