Home
last modified time | relevance | path

Searched refs:max_call_depth (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/
Dcontext.py222 def set_max_call_depth(self, max_call_depth): argument
223 if max_call_depth <= 0:
225 self.set_param(ms_ctx_param.max_call_depth, max_call_depth)
519 max_device_memory=str, print_file_path=str, enable_sparse=bool, max_call_depth=int,
/third_party/mindspore/tests/ut/python/ops/
Dtest_control_ops.py928 context.set_context(max_call_depth=60)
931 context.set_context(max_call_depth=old_max_call_depth)
963 context.set_context(max_call_depth=80)
967 context.set_context(max_call_depth=old_max_call_depth)
991 context.set_context(max_call_depth=2000)
996 context.set_context(max_call_depth=old_max_call_depth)
1028 context.set_context(max_call_depth=80)
1034 context.set_context(max_call_depth=old_max_call_depth)
/third_party/mindspore/tests/st/networks/models/bert/bert_performance/
Dtest_bert_thor.py148 context.set_context(max_call_depth=3000)
/third_party/mindspore/tests/st/networks/models/resnet50/src_thor/
Dthor.py149 context.set_context(max_call_depth=10000)
/third_party/mindspore/mindspore/nn/optim/
Dthor.py358 context.set_context(max_call_depth=10000)