Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/dataset/
Dtest_noop_mode.py26 context.set_ps_context(enable_ps=True)
33 context.set_ps_context(enable_ps=False)
38 context.set_ps_context(enable_ps=True)
45 context.set_ps_context(enable_ps=False)
/third_party/mindspore/tests/st/ps/cmp_sparse_embedding/
Dtest_cmp_sparse_embedding.py38 context.set_ps_context(enable_ps=True)
93 context.set_ps_context(enable_ps=True)
/third_party/mindspore/tests/st/ps/part_ps/
Dtest_ps_embedding_heterogeneous_conv2d_adam.py47 context.set_ps_context(enable_ps=True)
140 context.set_ps_context(enable_ps=False)
153 context.set_ps_context(enable_ps=True)
/third_party/mindspore/tests/st/ps/multi_full_ps/
Dtest_multi_full_ps.py34 context.set_ps_context(enable_ps=True)
/third_party/mindspore/tests/st/ps/full_ps/
Dtest_full_ps_lenet.py38 context.set_ps_context(enable_ps=True)
/third_party/mindspore/mindspore/
Dcontext.py839 def set_ps_context(**kwargs): function