Home
last modified time | relevance | path

Searched defs:dtype2 (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/aicpu/
Dsspaddmm.py37 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/
Dbinary_ext_op.cc38 auto dtype2 = input_args[kInputIndex1]->GetType(); in InferType() local
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/ops/primitive/
Dinplaceadd_ops.py17 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/
Dutils_const.py289 def _promote(dtype1, dtype2): argument
327 def _check_same_type(dtype1, dtype2): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Dmath_func.py7975 def _check_same_type(dtype1, dtype2): argument