Searched refs:const_bias (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/custom_ops_tbe/ |
| D | add3_impl.py | 24 def add3_compute(input1, input2, const_bias): argument
26 sum3 = te.lang.cce.vadds(sum2, tvm.const(const_bias, dtype=input1.dtype))
47 def CusAdd3Impl(input1, inptu2, sum1, const_bias, kernel_name="CusAdd3Impl"): argument
55 res = add3_compute(input1, input2, const_bias)
|
| D | cus_add3.py | 23 def __init__(self, const_bias=0.0): argument
|