Searched refs:mixed_precision_cast (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/model/ |
| D | test_mix_precision.py | 198 a = F.mixed_precision_cast(mstype.float16, tensor_a)
199 b = F.mixed_precision_cast(mstype.float16, tensor_b)
211 d = F.mixed_precision_cast(mstype.float16, tensor_c)
212 dict_cast = F.mixed_precision_cast(mstype.float16, kwargs)
|
| /third_party/mindspore/mindspore/train/ |
| D | amp.py | 103 label = F.mixed_precision_cast(mstype.float32, label)
104 return self._loss_fn(F.mixed_precision_cast(mstype.float32, out), label)
|
| /third_party/mindspore/mindspore/_extends/ |
| D | builtin_operations.py | 176 def mixed_precision_cast(dst_type, x): function
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 226 mixed_precision_cast = Primitive("mixed_precision_cast") variable
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | cell_wrapper.py | 598 label = F.mixed_precision_cast(mstype.float32, label)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_control_ops.py | 772 z = F.mixed_precision_cast(mstype.float16, x)
|