Searched refs:allow_dtypes (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 42 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
43 validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|
| D | image.py | 133 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
134 validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|
| D | embedding.py | 47 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
48 validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|
| D | rnns.py | 42 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
43 validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|
| D | math.py | 50 def _check_input_dtype(param_name, input_dtype, allow_dtypes, cls_name): argument
51 validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 168 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
169 Validator.check_type_name(param_name, input_dtype, allow_dtypes, cls_name)
|