Searched refs:scalar_mul (Results 1 – 7 of 7) sorted by relevance
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pipeline/parse/ |
D | parser_integrate.py | 32 return scalar_mul(g(x, x), g(x, x)) 42 return scalar_mul(g(x, y), h(x, y)) 44 return f((scalar_add, scalar_mul), a, b) 47 def scalar_mul(x, y): function 83 tup = scalar_add, scalar_mul 169 a = F.scalar_mul(x, x)
|
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/ir/ |
D | clone_test.py | 20 scalar_mul = Primitive(Constants.kScalarMul) variable 25 a = scalar_mul(x, x) 26 b = scalar_mul(y, y)
|
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/transform/ |
D | multi_relu_case.py | 25 scalar_mul = Primitive(Constants.kScalarMul) variable 29 z = scalar_mul(x, y)
|
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/optimizer/ad/ |
D | ad_test.py | 42 def scalar_mul(x, y): function 113 return scalar_mul(g(x, y), h(x, y)) 115 return f((scalar_add, scalar_mul), a, b) 212 tup = scalar_add, scalar_mul
|
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/optimizer/ |
D | opt_test.py | 31 scalar_mul = Primitive(Constants.kScalarMul) variable 38 return scalar_mul(10, scalar_add(5, 4)) 357 c = scalar_mul(a, b) 363 b = scalar_add(scalar_mul(a, y), scalar_div(a, x)) 364 c = scalar_add(scalar_mul(a, y), scalar_div(scalar_add(x, y), x)) 378 return scalar_mul(x, 0) 382 return scalar_mul(0, x) 390 return scalar_mul(x, 1) 394 return scalar_mul(1, x)
|
/third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
D | mul_impl.py | 39 return F.scalar_mul(x, y)
|
/third_party/mindspore/mindspore/ops/ |
D | functional.py | 194 scalar_mul = Primitive(_constants.kScalarMul) variable
|