Searched refs:func_single_output (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_pyfunc_op.py | 25 def func_single_output(x1, x2): function
53 expect = func_single_output(x1, x2)
56 net = PyFuncNet(func_single_output, [ms_dtype, ms_dtype], [shape, shape], [ms_dtype], [shape])
133 expect = func_single_output(x1, x2)
135 net = PyFuncGraph(func_single_output, [ms_dtype, ms_dtype], [shape, shape], [ms_dtype], [shape])
|