Searched defs:dtype2 (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/aicpu/ |
| D | sspaddmm.py | 37 def _reg_format(op_info, dtype1, dtype2, dtype3, alpha, beta): argument
42 def _reg_format_beta(op_info, dtype1, dtype2, alpha, beta): argument
53 def _reg_format_alpha(op_info, dtype1, dtype2, alpha): argument
69 def _reg_format_indices(op_info, dtype1, dtype2): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | binary_ext_op.cc | 38 auto dtype2 = input_args[kInputIndex1]->GetType(); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/ops/primitive/ |
| D | inplaceadd_ops.py | 17 def __init__(self, inputx_shape, inputv_shape, indices, dtype1=np.float32, dtype2=np.float32): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/numpy/ |
| D | utils_const.py | 289 def _promote(dtype1, dtype2): argument
327 def _check_same_type(dtype1, dtype2): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | math_func.py | 7975 def _check_same_type(dtype1, dtype2): argument
|