Searched defs:boost_level (Results 1 – 2 of 2) sorted by relevance
461 def _check_level(level, boost_level): argument542 def build_train_network(network, optimizer, loss_fn=None, level='O0', boost_level='O0', **kwargs): argument
347 amp_level="O0", boost_level="O0", **kwargs): argument