Searched defs:batch2 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/extend/ |
| D | math_func.py | 29 def baddbmm(input, batch1, batch2, beta=1, alpha=1): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_func_addbmm.py | 25 def construct(self, x, batch1, batch2, alpha=0.1, beta=0.5): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/ops/ |
| D | test_math_ops.py | 701 def construct(self, x, batch1, batch2): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | math_func.py | 4582 def addbmm(input, batch1, batch2, *, beta=1, alpha=1): argument
8345 def baddbmm(input, batch1, batch2, beta=1, alpha=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 3802 def addbmm(x, batch1, batch2, *, beta=1, alpha=1): argument
3851 def baddbmm(x, batch1, batch2, beta=1, alpha=1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1328 def addbmm(self, batch1, batch2, *, beta=1, alpha=1): argument
1378 def baddbmm(self, batch1, batch2, beta=1, alpha=1): argument
|