Searched refs:set_cache_enable (Results 1 – 9 of 9) sorted by relevance
94 set_cache_enable(false); in Reset()177 void PSContext::set_cache_enable(bool cache_enable) const { in set_cache_enable() function in mindspore::ps::PSContext179 PsDataPrefetch::GetInstance().set_cache_enable(cache_enable); in set_cache_enable()
81 void set_cache_enable(bool cache_enable) const;
38 … .def_property("cache_enable", &ParamInfo::cache_enable, &ParamInfo::set_cache_enable) in __anon2c0500310102()
38 EXPORT void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable() function
92 void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable() function
313 void set_cache_enable(bool cache_enable = true) { cache_enable_ = cache_enable; }
228 ps_context().set_cache_enable(cache_enable)
359 .def("set_cache_enable", &PSContext::set_cache_enable, "Set ps mode cache enable or not.") in PYBIND11_MODULE()
178 host_tensor->set_cache_enable(true); in BindAndInitCacheTensor()