Home
last modified time | relevance | path

Searched defs:scale_update_cell (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_loss_scale.py60 def __init__(self, network, optimizer, scale_update_cell=None): argument
/third_party/mindspore/tests/st/networks/models/bert/src/
Dbert_for_pre_training.py349 def __init__(self, network, optimizer, scale_update_cell=None): argument