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