Home
last modified time | relevance | path

Searched refs:set_backend_policy (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/
Dcontext.py182 self.set_backend_policy("vm")
186 self.set_backend_policy("ge")
192 def set_backend_policy(self, policy): member in _Context
193 success = self._context_handle.set_backend_policy(policy)
208 self.set_backend_policy("vm")
341 _k_context.set_backend_policy(default_backend)
/third_party/mindspore/tests/ut/python/parallel/
Dtest_comm_not_recompute.py74 _Context().set_backend_policy("vm")
83 _Context().set_backend_policy("ge")
/third_party/mindspore/tests/ut/cpp/pynative/
Dpynative_execute_test.cc76 ctx3->set_backend_policy("ge_only"); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/pybind_api/utils/
Dms_context_py.cc109 … .def("set_backend_policy", &mindspore::MsContext::set_backend_policy, "Set backend policy.") in __anon41857d0f0202()
/third_party/mindspore/mindspore/core/utils/
Dms_context.cc114 bool MsContext::set_backend_policy(const std::string &policy) { in set_backend_policy() function in mindspore::MsContext
Dms_context.h153 bool set_backend_policy(const std::string &policy);