Searched refs:allow_mix (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 576 … const std::string &prim_name, const bool allow_mix) { in CheckScalarOrTensorTypesSame() argument
577 auto arg_ = _CheckTypeSame(args, prim_name, allow_mix); in CheckScalarOrTensorTypesSame()
582 const bool allow_mix) { in _CheckTypeSame() argument
594 if (!allow_mix) { in _CheckTypeSame()
609 if (!allow_mix) { in _CheckTypeSame()
|
| D | check_convert_utils.h | 292 bool allow_mix = false);
322 const bool allow_mix);
|
| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 548 def check_scalar_or_tensor_types_same(args, valid_values, prim_name, allow_mix=False): argument
572 elif allow_mix:
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3144 … validator.check_scalar_or_tensor_types_same(args_lr, valid_dtypes, self.name, allow_mix=True)
3263 … validator.check_scalar_or_tensor_types_same(args_lr, valid_dtypes, self.name, allow_mix=True)
|