Searched defs:fp32_param (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/amp/ |
| H A D | amp.py | 83 def _ndarray_wrapper(f, target_dtype, fp32_param=None, cond_arg=None): argument
115 def _symbol_wrapper(f, target_dtype, fp32_param=None, cond_arg=None): argument
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/python/mxnet/contrib/amp/ |
| H A D | amp.py | 83 def _ndarray_wrapper(f, target_dtype, fp32_param=None, cond_arg=None): argument
115 def _symbol_wrapper(f, target_dtype, fp32_param=None, cond_arg=None): argument
|