Searched refs:ScalarMulAdd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 306 ALWAYS_INLINE void ScalarMulAdd(const float a, const float** inp, float** out) { in ScalarMulAdd() function
312 ALWAYS_INLINE void ScalarMulAdd(const float a, const bfloat16** inp, in ScalarMulAdd() function
744 ScalarMulAdd(sl1, &r, &out); in GEPP()
745 ScalarMulAdd(sl2, &nr, &nout); in GEPP()
746 ScalarMulAdd(sl3, &n2r, &n2out); in GEPP()
747 ScalarMulAdd(sl4, &n3r, &n3out); in GEPP()
761 ScalarMulAdd(sl, &r, &out); in GEPP()
|