H A D | sparse_matmul_op.cc | 314 const bfloat16** inp3, float** out) { 317 float inp3_f = ConvertBfloat16ToFloat(*inp3); 322 ++*inp3; 327 const float** inp2, const float** inp3, 329 **out += a1 * **inp1 + a2 * **inp2 + a3 * **inp3; 333 ++*inp3; 423 auto inp3 = reinterpret_cast<const float*>(*binp3); local 434 const auto b3 = LOAD(inp3); 456 auto inp3 = reinterpret_cast<const float*>(*binp3); local 461 const auto b3 = LOAD(inp3); 311 ScalarMulAdd3Way(const float a1, const float a2, const float a3, const bfloat16** inp1, const bfloat16** inp2, const bfloat16** inp3, float** out) argument 325 ScalarMulAdd3Way(const float a1, const float a2, const float a3, const float** inp1, const float** inp2, const float** inp3, float** out) argument 505 MulAdd3Way128(const Packet a1, const Packet a2, const Packet a3, const bfloat16** inp1, const bfloat16** inp2, const bfloat16** inp3, float** out) argument 526 MulAdd3Way(const Packet a1, const Packet a2, const Packet a3, const float** inp1, const float** inp2, const float** inp3, float** out) argument 544 TwoMulAdd3Way(const Packet a1, const Packet a2, const Packet a3, const float** inp1, const float** inp2, const float** inp3, float** out) argument 573 FourMulAdd3Way(const Packet a1, const Packet a2, const Packet a3, const float** inp1, const float** inp2, const float** inp3, float** out) argument 582 MulAdd3Way128(const Packet a1, const Packet a2, const Packet a3, const float** inp1, const float** inp2, const float** inp3, float** out) argument [all...] |